()
| 58 | |
| 59 | |
| 60 | def create_argparser(): |
| 61 | defaults = dict( |
| 62 | data_dir="", |
| 63 | schedule_sampler="uniform", |
| 64 | lr=1e-4, |
| 65 | weight_decay=0.0, |
| 66 | lr_anneal_steps=0, |
| 67 | batch_size=1, |
| 68 | microbatch=-1, # -1 disables microbatches |
| 69 | ema_rate="0.9999", # comma-separated list of EMA values |
| 70 | log_interval=10, |
| 71 | save_interval=10000, |
| 72 | resume_checkpoint="", |
| 73 | use_fp16=False, |
| 74 | fp16_scale_growth=1e-3, |
| 75 | ) |
| 76 | defaults.update(model_and_diffusion_defaults()) |
| 77 | parser = argparse.ArgumentParser() |
| 78 | add_dict_to_argparser(parser, defaults) |
| 79 | return parser |
| 80 | |
| 81 | |
| 82 | if __name__ == "__main__": |
no test coverage detected