()
| 73 | |
| 74 | |
| 75 | def create_argparser(): |
| 76 | defaults = dict( |
| 77 | data_dir="", |
| 78 | schedule_sampler="uniform", |
| 79 | lr=1e-4, |
| 80 | weight_decay=0.0, |
| 81 | lr_anneal_steps=0, |
| 82 | batch_size=1, |
| 83 | microbatch=-1, |
| 84 | ema_rate="0.9999", |
| 85 | log_interval=10, |
| 86 | save_interval=10000, |
| 87 | resume_checkpoint="", |
| 88 | use_fp16=False, |
| 89 | fp16_scale_growth=1e-3, |
| 90 | ) |
| 91 | defaults.update(sr_model_and_diffusion_defaults()) |
| 92 | parser = argparse.ArgumentParser() |
| 93 | add_dict_to_argparser(parser, defaults) |
| 94 | return parser |
| 95 | |
| 96 | |
| 97 | if __name__ == "__main__": |
no test coverage detected