Skip to content

[Trivial Fix] Flag use_ema in train_unconditional.py should not have default value #1654

@nkirschi

Description

@nkirschi

Problem

The definition of the flag use_ema wrongly includes a default value of True:

parser.add_argument(
"--use_ema",
action="store_true",
default=True,
help="Whether to use Exponential Moving Average for the final model weights.",
)

Hence, when the flag is not passed via CLI, it is still evaluated as true.

Fix

To fix this bug, simply remove default=True from the flag definition.

Metadata

Metadata

Assignees

Labels

staleIssues that haven't received updates

Type

No type
No fields configured for issues without a type.

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions