Train test on cpu

parent 831f063d
......@@ -13,7 +13,7 @@ if __name__ == "__main__":
batch_size=5,
num_epoch=10,
load_checkpoint=False,
context='gpu',
context='cpu',
normalize=True,
optimizer='adam',
optimizer_params={
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment