Loading model
(path, model=None, mode="evaluate")
| 252 | |
| 253 | |
| 254 | def load(path, model=None, mode="evaluate"): |
| 255 | """Loading model""" |
| 256 | if model is None: |
| 257 | with open(path + ".args", "rb") as f: |
| 258 | args = ArgConfig(["--None"]) |
| 259 | args.namespace = pickle.load(f) |
| 260 | model = Model(args) |
| 261 | model_state, _ = fluid.load_dygraph(path) |
| 262 | model.set_dict(model_state) |
| 263 | return model |