Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • rxsang's avatar
    Fix Resnet XLA with multi-GPUs (#6510) · 6dea4846
    rxsang authored 
    Don't pass `batch_size` to keras.layers.Input in DS multi-replica case. There is currently a bug in Keras side which will cause a batch size incompatible error.
    6dea4846
keras_imagenet_main.py 8.86 KB