ResNet training changes (update benchmark) (#2390)

* default arg for chunk

* bring back to_

* good changes

* new set

* unused hash

* fix optim

* new torch loader

* fix test lr scheduler
This commit is contained in:
George Hotz
2023-11-22 17:41:12 -08:00
committed by GitHub
parent 2dec86970a
commit cbb8486779
16 changed files with 132 additions and 30 deletions

1
.gitignore vendored
View File

@@ -43,3 +43,4 @@ temp
coverage.xml
htmlcov
outputs_yolov8
wandb