Saving model
(path, args, model, optimizer)
| 244 | |
| 245 | |
| 246 | def save(path, args, model, optimizer): |
| 247 | """Saving model""" |
| 248 | fluid.save_dygraph(model.state_dict(), path) |
| 249 | fluid.save_dygraph(optimizer.state_dict(), path) |
| 250 | with open(path + ".args", "wb") as f: |
| 251 | pickle.dump(args.namespace, f, protocol=2) |
| 252 | |
| 253 | |
| 254 | def load(path, model=None, mode="evaluate"): |