Skip to content

GitLab

Unverified Commit 990540b7 authored by Stas Bekman's avatar Stas Bekman Committed by GitHub
Browse files

[modelcard] fix (#12422) 

this PR is fixing an incorrect attribute - probably some tests are needed?
parent dc42e770
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/modelcard.py
View file @ 990540b7
...@@ -737,7 +737,7 @@ def extract_hyperparameters_from_trainer(trainer): ...@@ -737,7 +737,7 @@ def extract_hyperparameters_from_trainer(trainer):
if trainer.args.fp16: if trainer.args.fp16:
if trainer.use_amp: if trainer.use_amp:
hyperparameters["mixed_precision_training"] = "Native AMP" hyperparameters["mixed_precision_training"] = "Native AMP"
elif trainer._use_apex: elif trainer.use_apex:
hyperparameters["mixed_precision_training"] = f"Apex, opt level {trainer.args.fp16_opt_level}" hyperparameters["mixed_precision_training"] = f"Apex, opt level {trainer.args.fp16_opt_level}"
if trainer.args.label_smoothing_factor != 0.0: if trainer.args.label_smoothing_factor != 0.0:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment