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

Fix missing device arg in train.py, fix #2488

Browse Source
This commit is contained in:
Ross Wightman 2025-05-14 13:51:43 -07:00
parent 6b302f27a3
commit cabd26d018
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
train.py
View File

@ -892,6 +892,7 @@ def main():
optimizer, optimizer,
train_loss_fn, train_loss_fn,
args, args,
device=device,
lr_scheduler=lr_scheduler, lr_scheduler=lr_scheduler,
saver=saver, saver=saver,
output_dir=output_dir, output_dir=output_dir,