fjmscut

Results 3 comments of fjmscut

same bug 'xla.gpu.all_reduce' failed, i found that "tensorflow.keras.applications.inception_resnet_v2.InceptionResNetV2" was called before all_reduce, the problem occur, but if i use dense layer as replacement, it work well.... sb. help me plz

i guess the former batchnormalize layer called cause the later all_reduce crash as the code above

same bug when i implement gradient accumulate feature, sb. can help me ? plz .....