Xiao Zeqi commited on
Update exp_base.py
Browse files- experiments/exp_base.py +1 -1
experiments/exp_base.py
CHANGED
|
@@ -64,7 +64,7 @@ def load_custom_checkpoint(algo, checkpoint_path):
|
|
| 64 |
else:
|
| 65 |
filtered_state_dict[k] = v
|
| 66 |
|
| 67 |
-
algo.load_state_dict(filtered_state_dict, strict=
|
| 68 |
|
| 69 |
elif checkpoint_path.suffix == ".pt":
|
| 70 |
# Load from a .pt file
|
|
|
|
| 64 |
else:
|
| 65 |
filtered_state_dict[k] = v
|
| 66 |
|
| 67 |
+
algo.load_state_dict(filtered_state_dict, strict=False)
|
| 68 |
|
| 69 |
elif checkpoint_path.suffix == ".pt":
|
| 70 |
# Load from a .pt file
|