(experiment_directory, experiment_specs)
| 86 | |
| 87 | |
| 88 | def build_decoder(experiment_directory, experiment_specs): |
| 89 | |
| 90 | arch = __import__( |
| 91 | "networks." + experiment_specs["NetworkArch"], fromlist=["Decoder"] |
| 92 | ) |
| 93 | |
| 94 | latent_size = experiment_specs["CodeLength"] |
| 95 | |
| 96 | decoder = arch.Decoder(latent_size, **experiment_specs["NetworkSpecs"]).cuda() |
| 97 | |
| 98 | return decoder |
| 99 | |
| 100 | |
| 101 | def load_decoder( |