Return configuration.
(config_name="revnet-38", dataset="cifar-10")
| 108 | |
| 109 | |
| 110 | def get_config(config_name="revnet-38", dataset="cifar-10"): |
| 111 | """Return configuration.""" |
| 112 | print("Config: {}".format(config_name)) |
| 113 | sys.stdout.flush() |
| 114 | config = { |
| 115 | "revnet-38": config_.get_hparams_cifar_38(), |
| 116 | "revnet-110": config_.get_hparams_cifar_110(), |
| 117 | "revnet-164": config_.get_hparams_cifar_164(), |
| 118 | }[config_name] |
| 119 | |
| 120 | if dataset == "cifar-10": |
| 121 | config.add_hparam("n_classes", 10) |
| 122 | config.add_hparam("dataset", "cifar-10") |
| 123 | else: |
| 124 | config.add_hparam("n_classes", 100) |
| 125 | config.add_hparam("dataset", "cifar-100") |
| 126 | |
| 127 | return config |
| 128 | |
| 129 | |
| 130 | def get_datasets(data_dir, config): |
no test coverage detected