(model_g, model_c, load_path)
| 58 | torch.save(save_dic, save_path) |
| 59 | |
| 60 | def load_model(model_g, model_c, load_path): |
| 61 | checkpoint = torch.load(load_path) |
| 62 | model_g.load_state_dict(checkpoint['g_state_dict']) |
| 63 | model_c.load_state_dict(checkpoint['c_state_dict']) |
| 64 | return model_g, model_c |
nothing calls this directly
no outgoing calls
no test coverage detected