update eps

Janetalready · Sep 21, 2022 · bfee69a · bfee69a
1 parent a988b48
commit bfee69a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train_dreamer_add_class_gen.py b/train_dreamer_add_class_gen.py
@@ -330,7 +330,7 @@ def train(opt):
 
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     sender = Sender(device, opt.max_step, opt, width = 128)
-    receiver = ReceiverOnestep(device, opt.game_size, 128, opt, eps=opt.eps)
+    receiver = ReceiverOnestep(device, opt.game_size, 128, opt)
     value_model = ValueModel(128, opt.game_size)
 
     sender.to(device)