This PR exposes three arguments from the NCSN++ backbone to the CLI, allowing to easily change them when invoking the training script (e.g. to create a smaller variant of the model). It also removes a spurious add_argparse_args method and puts the new method at the top of the class as in other places in the codebase.
This PR exposes three arguments from the NCSN++ backbone to the CLI, allowing to easily change them when invoking the training script (e.g. to create a smaller variant of the model). It also removes a spurious
add_argparse_args
method and puts the new method at the top of the class as in other places in the codebase.