Commit 6ed01422 authored by Bharath Ramsundar's avatar Bharath Ramsundar Committed by GitHub
Browse files

Merge pull request #871 from miguelroboso/master 

Set model_dir before restoring tensorgraph in load_from_dir
parents 98eba0bb 29d601a7

deepchem/models/tensorgraph/tensor_graph.py

+1 −0
Original line number Diff line number Diff line
@@ -780,6 +780,7 @@ class TensorGraph(Model): 
    with open(pickle_name, 'rb') as fout:

      tensorgraph = pickle.load(fout)

      tensorgraph.built = False

      tensorgraph.model_dir = model_dir

      try:

        tensorgraph.restore()

      except ValueError:

CRA Git | Maintained and supported by SUSTech CRA and CCSE