Skip to content

Instantly share code, notes, and snippets.

@piercelamb

piercelamb/test_data.py

Created December 20, 2022 17:58
Show Gist options
  • Save piercelamb/8538fe09042e62cc39a7998f4f3df176 to your computer and use it in GitHub Desktop.
Save piercelamb/8538fe09042e62cc39a7998f4f3df176 to your computer and use it in GitHub Desktop.
Download ZIP
test_data
Raw
test_data.py
test_data = load_from_disk(SAGEMAKER_LOCAL_INFERENCE_DATA_DIR)
print(f"Running inference over {test_data.num_rows} samples")
test_data.set_format(type="torch")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment