Commit 29d601a7 authored by michele donato's avatar michele donato
Browse files

Set model_dir before restoring tensorgraph in load_from_dir

parent 908a0919

deepchem/models/tensorgraph/tensor_graph.py

+1 −0
Original line number Diff line number Diff line
@@ -679,6 +679,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