xla
[SPMD] input sharding for both train & test data loading in imagenet example
#6515
Merged
Merged
[SPMD] input sharding for both train & test data loading in imagenet example #6515
yeounoh
merged
1 commit into
master
from
update_spmd_example
Input sharding should be applied to both train and test data
4782b066
vanbasten23
approved these changes
on 2024-02-10
Login to write a write a comment.
Login via GitHub