Views
No views yet
| File | Description | Size |
|---|---|---|
autoencoder_small_best_epoch_050.pt | Full autoencoder (encoder + decoder) | ~900KB |
autoencoder_small_decoder_epoch_050.pt | Decoder only (for inference) | ~150KB |
agent_epoch_00100.pt | World model checkpoint (epoch 100) | ~83MB |
agent_epoch_00125.pt | World model checkpoint (epoch 125) | ~83MB |
agent_epoch_00150.pt | World model checkpoint (epoch 150) | ~83MB |
| Index | Action |
|---|---|
| 0 | idle |
| 1 | attack |
| 2 | block |
1# Load world model
2import torch
3agent = torch.load("agent_epoch_00150.pt")
4
5# Load decoder for upsampling
6decoder = torch.load("autoencoder_small_decoder_epoch_050.pt")