()
| 200 | |
| 201 | |
| 202 | def create_argparser(): |
| 203 | defaults = dict( |
| 204 | data_dir="", |
| 205 | val_data_dir="", |
| 206 | noised=True, |
| 207 | iterations=150000, |
| 208 | lr=3e-4, |
| 209 | weight_decay=0.0, |
| 210 | anneal_lr=False, |
| 211 | batch_size=4, |
| 212 | microbatch=-1, |
| 213 | schedule_sampler="uniform", |
| 214 | resume_checkpoint="", |
| 215 | log_interval=10, |
| 216 | eval_interval=5, |
| 217 | save_interval=10000, |
| 218 | ) |
| 219 | defaults.update(classifier_and_diffusion_defaults()) |
| 220 | parser = argparse.ArgumentParser() |
| 221 | add_dict_to_argparser(parser, defaults) |
| 222 | return parser |
| 223 | |
| 224 | |
| 225 | if __name__ == "__main__": |
no test coverage detected