Use pytorch-lightning for better readability and optimization
EmmanuelMess opened this issue · comments
pytorch-lightning allows for less boilerplate and more optimization. Maybe it should be used to allow for easier reuse of the code.
seungeunrho / minimalRL
Implementations of basic RL algorithms with minimal lines of codes! (pytorch based)