Xiao Zeqi commited on
Commit
ac0e98a
·
unverified ·
1 Parent(s): 5555c7b

Update exp_base.py

Browse files
Files changed (1) hide show
  1. 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=True)
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