search

huawei-noah / xingtian

xingtian is a componentized library for the development and verification of reinforcement learning algorithms
MIT License
301 stars 89 forks source link

关于使用yaml自定义模型 #2

Closed hecong97 closed 3 years ago

hecong97 commented 3 years ago

1、关于使用yaml文件参数自定义模型，如果可以自定义卷积层或者全连接的层数易用性应该会更好一些？ 2、好像没有实现lstm模块

hustqj commented 3 years ago

现在的确没有通过参数定义卷积或全连接层数，目前比较推荐用户自己定义模型文件，然后在yaml文件里面引用即可，lstm模块也可用户自己定义