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

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions