xla
Enable imagenet AMP optimzer test in circleci for GPU test
#3233

Merged

Enable imagenet AMP optimzer test in circleci for GPU test #3233

JackCaoG merged 2 commits into master from enable_imagenet_amp_test_on_gpu

yeounoh3 years ago

Addressing the comment from #3228 (comment)

Enable imagenet AMP optimzer test in circleci for GPU test

4dfa2347

yeounoh requested a review from

JackCaoG 3 years ago

yeounoh assigned

yeounoh 3 years ago

JackCaoG commented on 2021-12-02

.circleci/common.sh

	125		echo "Running ImageNet Test"
	126		python test/test_train_mp_imagenet_amp.py --fake_data --num_epochs=1
	127
	128		# disabled per https://github.com/pytorch/xla/pull/2810

JackCaoG3 years ago

Should be #2809

yeounoh3 years ago

Updated the comment, thanks!

Enable imagenet AMP optimzer test in circleci for GPU test

ee5383ee

JackCaoG approved these changes on 2021-12-02

JackCaoG3 years ago

Thanks!

wonjoo-wj approved these changes on 2021-12-03

JackCaoG merged cfe4711d into master 3 years ago

JackCaoG deleted the enable_imagenet_amp_test_on_gpu branch 3 years ago

Reviewers

wonjoo-wj

JackCaoG

Assignees

yeounoh

Labels

None yet

Milestone

No milestone