浏览代码

Temporarily disable exporting

/comms-grad
Ervin Teng 4 年前
当前提交
3475c686
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 2
      ml-agents/mlagents/trainers/model_saver/torch_model_saver.py

2
ml-agents/mlagents/trainers/model_saver/torch_model_saver.py


}
torch.save(state_dict, f"{checkpoint_path}.pt")
torch.save(state_dict, os.path.join(self.model_path, "checkpoint.pt"))
self.export(checkpoint_path, behavior_name)
# self.export(checkpoint_path, behavior_name)
return checkpoint_path
def export(self, output_filepath: str, behavior_name: str) -> None:

正在加载...
取消
保存