Interface for U.make_session changed (#865)

This commit is contained in:
Sridhar Thiagarajan
2019-04-01 16:24:02 -07:00
committed by pzhokhov
parent 62a9c76f18
commit 6d1c6c78d3

View File

@@ -23,7 +23,7 @@ def model(inpt, num_actions, scope, reuse=False):
if __name__ == '__main__':
with U.make_session(8):
with U.make_session(num_cpu=8):
# Create the environment
env = gym.make("CartPole-v0")
# Create all the functions necessary to train the model