Skip to content
Snippets Groups Projects
Commit ad1477c7 authored by David Carreto Fidalgo's avatar David Carreto Fidalgo
Browse files

set log_level and find_unused_parameters

parent aef2f8bb
Branches
No related tags found
No related merge requests found
......@@ -47,6 +47,8 @@ def main():
evaluation_strategy="epoch",
learning_rate=2e-5,
weight_decay=0.01,
log_level="info",
ddp_find_unused_parameters=False,
)
trainer = Trainer(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment