Flux.jl
Add a ResNet-18 / Tiny-ImageNet-200 example
#2696
Merged

Add a ResNet-18 / Tiny-ImageNet-200 example #2696

CarloLucibello merged 7 commits into master from cl/resnet-tinyimagenet
CarloLucibello
CarloLucibello Add ResNet-18 / Tiny-ImageNet training example
e786d9bc
CarloLucibello examples/resnet_tinyimagenet: rely on train! memory defaults
f6fcdacf
CarloLucibello add pytorch example
94879707
CarloLucibello CarloLucibello force pushed from ee6816f9 to 94879707 75 days ago
CarloLucibello examples/resnet_tinyimagenet: add seed, LR schedule, and precision/re…
089a0655
CarloLucibello examples/resnet_tinyimagenet/pytorch: use device iterator in flux_bench
e88545ff
CarloLucibello cleanup
21936763
CarloLucibello examples/resnet_tinyimagenet: remove bfloat16 support
e9994fad
CarloLucibello 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
CarloLucibello CarloLucibello merged bad1d314 into master 75 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone