Skip to content

Instantly share code, notes, and snippets.

@Tob-iee
Last active October 25, 2023 16:57
Show Gist options
  • Save Tob-iee/8b6f75c26b39a1fb1e542f70ee472584 to your computer and use it in GitHub Desktop.
Save Tob-iee/8b6f75c26b39a1fb1e542f70ee472584 to your computer and use it in GitHub Desktop.
Download the training dataset versions attached to experiment
attached_datasets = experiment.list_attached_dataset_versions()
train_dataset_version = experiment.get_dataset(train_version_name)
eval_dataset_version = experiment.get_dataset(eval_version_name)
images_dir = os.path.join(os.getcwd(), train_set_local_dir)
if os.path.isdir(train_set_local_dir):
print("dataset version has already been downloaded")
else:
train_dataset_version.download(train_set_local_dir)
if os.path.isdir(eval_set_local_dir):
print("dataset version has already been downloaded")
else:
eval_dataset_version.download(eval_set_local_dir)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment