DeepSpeed
9bc4cd01 - Store/Load CIFAR from local/offline (#6390)

Comment changes are shownComment changes are hidden
Commit
291 days ago
Store/Load CIFAR from local/offline (#6390) CIFAR10_DATASET_PATH -> Path where the dataset is stored STORE_CIFAR10 -> Store the dataset 1/0 CIFAR10_OFFLINE -> To use offline dataset 1/0 MISC: Added getDeviceId to get device if by name in case of accelerator --------- Co-authored-by: Shaik Raza Sikander <srsikander@habana.ai> Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com> Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com> Co-authored-by: Logan Adams <loadams@microsoft.com>
Author
Parents
  • tests/unit
    • File
      alexnet_model.py