Skip to content

Instantly share code, notes, and snippets.

@shashankprasanna
Created March 14, 2020 00:18
Show Gist options
  • Save shashankprasanna/ee1d9d144f95bb26fc92190df5daa67f to your computer and use it in GitHub Desktop.
Save shashankprasanna/ee1d9d144f95bb26fc92190df5daa67f to your computer and use it in GitHub Desktop.
train_path = f's3://{bucket_name}/{dataset_folder}/cifar10-dataset/train'
val_path = f's3://{bucket_name}/{dataset_folder}/cifar10-dataset/validation'
eval_path = f's3://{bucket_name}/{dataset_folder}/cifar10-dataset/eval'
hvd_estimator.fit({'train': train_path,'validation': val_path,'eval': eval_path},
job_name=job_name, wait=False)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment