ResNet optimized #4491
chandrasekhard2
wants to merge
28 commits into
master
from
chandrasekhard-resnet50
ResNet optimized
9499d865
add cpu_to_device_transfer_threads
c5790bf8
JackCaoG
commented
on 2023-01-24
fix type
0af201bb
Added LARS optimizer
62b966b3
Added LARS implementation with multi-epoch loader
90b10f36
Add linear scaling, warmup epochs and poly decay. correct few LARS st…
ae042496
remove pjrt/test_train_pjrt_imagenet.py, revert args_parse, fix a pat…
2475d110
chandrasekhard2
commented
on 2023-02-16
Separated ResNet and added space_to_depth optimization
61c3ed7d
remove unncesessary functions
e519a866
Updated code to use shuffle after every epoch and code cleanup
eb3beff2
Updated resnet code
3c3d0d82
Update test_resnet.py
c811dae0
Update files
c9fcbeae
update code
7a153778
Update test_resnet.py
91298df5
Update test_resnet.py
3770297f
Update test_resnet.py
d1476b91
Update README.md
14982817
Update README.md
f7e0b6c2
Update README.md
de5db83f
Update README.md
e878709f
update instructions
851def71
Add v4_128 config
85b01050
update batch size to 512 because of tracing issue
1f7c1072
Update instructions
d04909ef
update instructions
70505525
Remove unnecessary files
f5778e31
update hyperparams
2d0001bc
Login to write a write a comment.
Login via GitHub