浏览代码

reverted .tf to .bytes into the policy.py script

/develop-generalizationTraining-TrainerController
vincentpierre 6 年前
当前提交
47de43f6
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 2
      ml-agents/mlagents/trainers/policy.py

2
ml-agents/mlagents/trainers/policy.py


input_binary=True,
input_checkpoint=ckpt.model_checkpoint_path,
output_node_names=target_nodes,
output_graph=(self.model_path + '.tf'),
output_graph=(self.model_path + '.bytes'),
clear_devices=True, initializer_nodes='', input_saver='',
restore_op_name='save/restore_all',
filename_tensor_name='save/Const:0')

正在加载...
取消
保存