(
experiment_directory, experiment_specs, checkpoint, data_parallel=True
)
| 99 | |
| 100 | |
| 101 | def load_decoder( |
| 102 | experiment_directory, experiment_specs, checkpoint, data_parallel=True |
| 103 | ): |
| 104 | |
| 105 | decoder = build_decoder(experiment_directory, experiment_specs) |
| 106 | |
| 107 | if data_parallel: |
| 108 | decoder = torch.nn.DataParallel(decoder) |
| 109 | |
| 110 | epoch = load_model_parameters(experiment_directory, checkpoint, decoder) |
| 111 | |
| 112 | return (decoder, epoch) |
| 113 | |
| 114 | |
| 115 | def load_latent_vectors(experiment_directory, checkpoint): |
nothing calls this directly
no test coverage detected