Skip to content

Commit 24bf09b

Browse files
committed
added tesnorflow to requirements
1 parent 90a2fc8 commit 24bf09b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

agents/Base_Agent.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
import numpy as np
77
import torch
88
import time
9+
import tensorflow as tf
910
from nn_builder.pytorch.NN import NN
1011
# from tensorboardX import SummaryWriter
1112
from torch.optim import optimizer
@@ -364,4 +365,4 @@ def move_gradients_one_model_to_another(from_model, to_model, set_from_gradients
364365
def copy_model_over(from_model, to_model):
365366
"""Copies model parameters from from_model to to_model"""
366367
for to_model, from_model in zip(to_model.parameters(), from_model.parameters()):
367-
to_model.data.copy_(from_model.data.clone())
368+
to_model.data.copy_(from_model.data.clone())

0 commit comments

Comments
 (0)