added tesnorflow to requirements

ranpku · Jul 18, 2019 · 24bf09b · 24bf09b
1 parent 90a2fc8
commit 24bf09b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/agents/Base_Agent.py b/agents/Base_Agent.py
@@ -6,6 +6,7 @@
 import numpy as np
 import torch
 import time
+import tensorflow as tf
 from nn_builder.pytorch.NN import NN
 # from tensorboardX import SummaryWriter
 from torch.optim import optimizer
@@ -364,4 +365,4 @@ def move_gradients_one_model_to_another(from_model, to_model, set_from_gradients
     def copy_model_over(from_model, to_model):
         """Copies model parameters from from_model to to_model"""
         for to_model, from_model in zip(to_model.parameters(), from_model.parameters()):
-            to_model.data.copy_(from_model.data.clone())
+            to_model.data.copy_(from_model.data.clone())