浏览代码

[skip ci] clean up around initializer

/trainer-plugin
Anupam Bhatnagar 4 年前
当前提交
5d9c110f
共有 1 个文件被更改,包括 5 次插入5 次删除
  1. 10
      ml-agents/mlagents/trainers/learn.py

10
ml-agents/mlagents/trainers/learn.py


}
print(f"plugins available {discovered_plugins}")
all_initializers = set(get_all_subclasses(Initializer))
all_initializers = list(get_all_subclasses(Initializer))
print("Registering new initializer")
print(f"Found {len(all_initializers)} new initializers")
distributed_init = all_initializers[0]()
distributed_init.load()
new_initializers = all_initializers - original_initializers
print(f"Found {len(new_initializers)} new initializers")
print(all_initializers)
print(new_initializers)
print(all_trainers)
print(new_trainers)

正在加载...
取消
保存