self.saver.register(self.policy)
self.saver.register(self.optimizer)
self.saver.initialize_or_load(self.policy)
self.saver.initialize_or_load()
# Needed to resume loads properly
self.step = policy.get_current_step()
NetworkSettings.MemorySettings() if use_rnn else None
)
policy = TorchPolicy(
0,
mock_behavior_specs,
trainer_config,
"test",
False,
tanhresample,
0, mock_behavior_specs, trainer_config, tanhresample, tanhresample
bc_module = BCModule(
policy,