pytorch/xla

Pull Requests Commits

update hyperparams

chandrasekhard2 committed 2 years ago

2d0001bc

Remove unnecessary files

chandrasekhard2 committed 2 years ago

f5778e31

update instructions

chandrasekhard2 committed 2 years ago

70505525

Update instructions

chandrasekhard2 committed 2 years ago

d04909ef

update batch size to 512 because of tracing issue

chandrasekhard2 committed 2 years ago

1f7c1072

Add v4_128 config

chandrasekhard2 committed 2 years ago

85b01050

update instructions

chandrasekhard2 committed 2 years ago

851def71

Update README.md

chandrasekhard2 committed 2 years ago

Verified e878709f

Update README.md

chandrasekhard2 committed 2 years ago

Verified de5db83f

Update README.md

chandrasekhard2 committed 2 years ago

Verified f7e0b6c2

Update README.md

chandrasekhard2 committed 2 years ago

Verified 14982817

Update test_resnet.py

chandrasekhard2 committed 2 years ago

Verified d1476b91

Update test_resnet.py

chandrasekhard2 committed 2 years ago

Verified 3770297f

Update test_resnet.py

chandrasekhard2 committed 2 years ago

Verified 91298df5

chandrasekhard2 committed 2 years ago

7a153778

chandrasekhard2 committed 2 years ago

c9fcbeae

Update test_resnet.py

chandrasekhard2 committed 2 years ago

Verified c811dae0

Updated resnet code

chandrasekhard2 committed 2 years ago

3c3d0d82

Updated code to use shuffle after every epoch and code cleanup

chandrasekhard2 committed 2 years ago

eb3beff2

remove unncesessary functions

chandrasekhard2 committed 2 years ago

e519a866

Separated ResNet and added space_to_depth optimization

chandrasekhard2 committed 2 years ago

61c3ed7d

remove pjrt/test_train_pjrt_imagenet.py, revert args_parse, fix a path in test_train_mp_imagenet.py

chandrasekhard2 committed 2 years ago

2475d110

Add linear scaling, warmup epochs and poly decay. correct few LARS steps and add feature to tune flags

chandrasekhard2 committed 2 years ago

ae042496

Added LARS implementation with multi-epoch loader

chandrasekhard2 committed 2 years ago

90b10f36

Added LARS optimizer

chandrasekhard2 committed 2 years ago

62b966b3

chandrasekhard2 committed 2 years ago

0af201bb

add cpu_to_device_transfer_threads

chandrasekhard2 committed 2 years ago

c5790bf8

ResNet optimized

chandrasekhard2 committed 2 years ago

9499d865

Revert "Revert "Adapt to new pytorch grad clipping API (#4480)" (#4484)" (#4486)

wonjoo-wj committed 2 years ago

Verified aff35132

Revert "Adapt to new pytorch grad clipping API (#4480)" (#4484)

wonjoo-wj committed 2 years ago

Verified 3e6bbbbd

Older