update directory structure

2023-05-30 17:16:48 +08:00
parent 4968e6400e
commit 50111160e3
37 changed files with 150 additions and 159 deletions
@@ -1,8 +1,8 @@
 import torch
 from tqdm import tqdm

-from tugo.agent.pg import PolicyAgent
-from tugo.agent.predict import DeepLearningAgent
+from tugo.agents.policy_gradient_agent import PolicyAgent
+from tugo.agents.predict import DeepLearningAgent
 from tugo.encoders.alphago import AlphaGoEncoder
 from tugo.rl.simulate import experience_simulation