mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Add device arg to validate() calls in train.py

Browse Source
This commit is contained in:
Ross Wightman 2024-02-04 10:14:57 -08:00
parent a08b57e801
commit c7ac37693d
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -881,6 +881,7 @@ def main():
loader_eval, loader_eval,
validate_loss_fn, validate_loss_fn,
args, args,
device=device,
amp_autocast=amp_autocast, amp_autocast=amp_autocast,
) )
@ -893,6 +894,7 @@ def main():
loader_eval, loader_eval,
validate_loss_fn, validate_loss_fn,
args, args,
device=device,
amp_autocast=amp_autocast, amp_autocast=amp_autocast,
log_suffix=' (EMA)', log_suffix=' (EMA)',
) )