refactor

README.md

@@ -34,7 +34,7 @@ More details about the project and the trained agent can be found in the [projec
 import gym
 import yaml
 import torch
-from helpers.load import QuantumNet
+from helpers.qnn import QuantumNet
 from helpers.wrappers import BinaryWrapper
 from helpers.agent import Agent
 
@@ -45,7 +45,7 @@ env = BinaryWrapper(env)
 
 # Network
 with open('config.yaml', 'r') as f:
-    hparams = yaml.safe_load(f)
+  hparams = yaml.safe_load(f)
 
 net = QuantumNet(hparams['n_layers'])
 state_dict = torch.load('qdqn-FrozenLake-v1.pt', map_location=torch.device('cpu'))

example.py

@@ -1,7 +1,7 @@
 import gym
 import yaml
 import torch
-from helpers.load import QuantumNet
+from helpers.qnn import QuantumNet
 from helpers.wrappers import BinaryWrapper
 from helpers.agent import Agent