moving options policy to policies, commenting options image, and making the calls to train more flexible

2021-10-21 05:59:28 -07:00
parent 45a99978e4
commit 05b31092f4
3 changed files with 174 additions and 48 deletions
--- a/src/policies/init.py
+++ b/src/policies/init.py
@@ -1 +1,2 @@
 from src.policies.policy import IntersimPolicy, IntersimStateNet, IntersimStateActionNet, generate_transforms
+from src.policies.options import OptionsCnnPolicy