(net, optimizer, model_path, epoch_id, prefix='rec')
| 23 | |
| 24 | |
| 25 | def save_model(net, optimizer, model_path, epoch_id, prefix='rec'): |
| 26 | model_path = os.path.join(model_path, str(epoch_id)) |
| 27 | _mkdir_if_not_exist(model_path) |
| 28 | model_prefix = os.path.join(model_path, prefix) |
| 29 | paddle.save(net.state_dict(), model_prefix + ".pdparams") |
| 30 | paddle.save(optimizer.state_dict(), model_prefix + ".pdopt") |
| 31 | logger.info("Already save model in {}".format(model_path)) |
| 32 | |
| 33 | |
| 34 | def save_jit_model(net, model_path, prefix='tostatic'): |
no test coverage detected