Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Set default models to training mode in the train_step. #103

Merged
merged 2 commits into from
Oct 23, 2024

Looks like moving the model.train() and model.eval() here should work.

411de9d
Select commit
Loading
Failed to load commit list.
Merged

Set default models to training mode in the train_step. #103

Looks like moving the model.train() and model.eval() here should work.
411de9d
Select commit
Loading
Failed to load commit list.
Codecov / codecov/patch failed Oct 23, 2024 in 1s

0.00% of diff hit (target 34.83%)

View this Pull Request on Codecov

0.00% of diff hit (target 34.83%)

Annotations

Check warning on line 141 in src/fibad/pytorch_ignite.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

src/fibad/pytorch_ignite.py#L141

Added line #L141 was not covered by tests

Check warning on line 182 in src/fibad/pytorch_ignite.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

src/fibad/pytorch_ignite.py#L182

Added line #L182 was not covered by tests