()
| 247 | |
| 248 | |
| 249 | def create_argparser(): |
| 250 | defaults = dict( |
| 251 | data_dir="", |
| 252 | val_data_dir="", |
| 253 | noised=True, |
| 254 | iterations=150000, |
| 255 | lr=3e-4, |
| 256 | weight_decay=0.0, |
| 257 | anneal_lr=False, |
| 258 | batch_size=4, |
| 259 | microbatch=-1, |
| 260 | schedule_sampler="uniform", |
| 261 | resume_checkpoint="", |
| 262 | log_interval=1, |
| 263 | eval_interval=1000, |
| 264 | save_interval=5000, |
| 265 | dataset='brats' |
| 266 | ) |
| 267 | defaults.update(classifier_and_diffusion_defaults()) |
| 268 | parser = argparse.ArgumentParser() |
| 269 | add_dict_to_argparser(parser, defaults) |
| 270 | return parser |
| 271 | |
| 272 | |
| 273 | if __name__ == "__main__": |
no test coverage detected