Gpu train - #941

Open

michaelzhiluo wants to merge 22 commits into

tensorflow:masterfrom

concretevitamin:gpu_train

Commits on Jul 24, 2021

Minor change
concretevitamin
committed
8827288

Commits on Jul 27, 2021

Multi-GPU Training
Ubuntu
committed
6544625
Delete run.log
michaelzhiluo
authored
41a7b12
Delete run.log
michaelzhiluo
authored
2e08e0e

Commits on Aug 3, 2021

okie
Ubuntu
committed
9e40451
ok
Ubuntu
committed
3017e06
ok
Ubuntu
committed
74aadab

Commits on Aug 6, 2021

temp
Ubuntu
committed
863f5c6
lmao
Ubuntu
committed
349f196
ok
Ubuntu
committed
9f7d4c1
Turn off amp
Ubuntu
committed
50c0fc6
enable amp
Ubuntu
committed
c875534

Commits on Aug 9, 2021

Make AMP work.
concretevitamin
committed
8a68cc5

Commits on Aug 10, 2021

Added static loss + Nvidia config
Ubuntu
committed
09199b8
ok
Ubuntu
committed
222cc86

Commits on Sep 7, 2021

GPU Inference
Ubuntu
committed
fb36b0e

Commits on Oct 7, 2021

Added requirements
Ubuntu
committed
19e6496
Merge branch 'gpu_train' of https://github.com/concretevitamin/tpu into gpu_train
Ubuntu
committed
b14bfb0

Commits on Oct 8, 2021

Small c hange
michaelzhiluo
committed
adf2a26
temp
Ubuntu
committed
9459fee
Merge branch 'gpu_train' of https://github.com/concretevitamin/tpu into gpu_train
Ubuntu
committed
bbe07fe

Commits on Sep 12, 2022

Add Resnet inference
michaelzhiluo
committed
d01593d