Bugfixes in valuedice

2021-08-04 20:53:03 +02:00
parent 2224e2cd14
commit 40c55478f3
8 changed files with 71 additions and 33 deletions
--- a/src/util/nn_training.py
+++ b/src/util/nn_training.py
@@ -1,3 +1,5 @@
+import torch
+
 def optimizer_factory(config, parameters):
    optimizer_type = config['optimizer']
    learning_rate = config['lr']