Add a ResNet-18 / Tiny-ImageNet-200 example #2696
Add ResNet-18 / Tiny-ImageNet training example
e786d9bc
examples/resnet_tinyimagenet: rely on train! memory defaults
f6fcdacf
add pytorch example
94879707
examples/resnet_tinyimagenet: add seed, LR schedule, and precision/re…
089a0655
examples/resnet_tinyimagenet/pytorch: use device iterator in flux_bench
e88545ff
cleanup
21936763
examples/resnet_tinyimagenet: remove bfloat16 support
e9994fad
CarloLucibello
changed the title Adaptive GPU memory management in train! (+ ResNet/Tiny-ImageNet example) Add a ResNet-18 / Tiny-ImageNet-200 example 75 days ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub