Views
No views yet
best_checkpoint.pt: PyTorch checkpoint (1180.4 MB)1import torch
2from aesop.models.policy import AesopPolicy
3
4# Load checkpoint
5checkpoint = torch.load("warm_up_best.pt", map_location="cpu")
6model = AesopPolicy()
7model.load_state_dict(checkpoint['model_state_dict'])1{
2 "stage": "warm_up",
3 "keys": [
4 "epoch",
5 "global_step",
6 "best_val_acc",
7 "model_state_dict",
8 "tactic_pool",
9 "wandb_run_id",
10 "random_seed",
11 "random_state",
12 "numpy_random_state",
13 "torch_rng_state",
14 "global_moving_avg",
15 "global_moving_std",
16 "optimizer_0_state_dict",
17 "optimizer_1_state_dict",
18 "optimizer_2_state_dict",
19 "cuda_rng_state",
20 "scaler_state_dict",
21 "trainer_state_dict",
22 "metrics"
23 ],
24 "epoch": 12,
25 "best_val_acc": 0.3483606557377049,
26 "model_architecture": {
27 "is_valid": true,
28 "components": [
29 {
30 "name": "shared_mlp",
31 "found_keys": 3,
32 "total_expected": 3
33 },
34 {
35 "name": "lemma_heads",
36 "found_keys": 2,
37 "total_expected": 2
38 },
39 {
40 "name": "config_heads",
41 "found_keys": 2,
42 "total_expected": 2
43 },
44 {
45 "name": "base_encoder",
46 "found_keys": 208,
47 "total_expected": "variable"
48 }
49 ],
50 "warnings": [
51 "Component 'tactic_heads' not found in state_dict"
52 ],
53 "errors": []
54 },
55 "file_size_mb": 1180.4070987701416,
56 "files_description": "- `best_checkpoint.pt`: PyTorch checkpoint (1180.4 MB)"
57}