Skip to content

Commit d737776

Browse files
authored
Remove explicit strict checkpoint loading in alf.bin.play (#1794)
1 parent c40bcd3 commit d737776

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

alf/trainers/policy_trainer.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1119,17 +1119,15 @@ def play(root_dir,
11191119
ignored_parameter_prefixes=ignored_parameter_prefixes,
11201120
including_optimizer=False,
11211121
including_replay_buffer=False,
1122-
including_data_transformers=True,
1123-
strict=True)
1122+
including_data_transformers=True)
11241123
except FileNotFoundError as e:
11251124
if checkpoint_step == 'best':
11261125
recovered_global_step = checkpointer.load(
11271126
'latest',
11281127
ignored_parameter_prefixes=ignored_parameter_prefixes,
11291128
including_optimizer=False,
11301129
including_replay_buffer=False,
1131-
including_data_transformers=True,
1132-
strict=True)
1130+
including_data_transformers=True)
11331131
else:
11341132
raise e
11351133
# The behavior of some algorithms is based by scheduler using training

0 commit comments

Comments
 (0)