overview
This solution enables you to:
- Train a PyTorch image classification model using transfer learning from torchvision models like ResNet and VGG on GPUs, leveraging Snowpark Container Services in Snowflake.
- Classify chest X-rays to detect the presence of pneumonia.
- Log, manage, and deploy the model through the Snowpark Model Registry for inference using Snowflake Warehouses (CPUs).
- Perform inference directly within Snowflake, with a front-end app built using Streamlit in Snowflake.