Skip to content

Commit

Permalink
Update ImageClassificationInSpark.ipynb
Browse files Browse the repository at this point in the history
bradmiro authored Dec 18, 2024
1 parent 8c83910 commit 4139a99
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion notebooks/ai-ml/ImageClassificationInSpark.ipynb
Original file line number Diff line number Diff line change
@@ -267,7 +267,7 @@
"collapsed": false
},
"source": [
"Create a [Pandas UDF](https://spark.apache.org/docs/3.4.2/api/python/reference/pyspark.sql/api/pyspark.sql.functions.pandas_udf.html) that contains the Torch code that will run on each worker to perform model inference. This training job uses the [ResNet50](https://pytorch.org/vision/main/models/generated/torchvision.models.resnet50.html) model."
"Create a [Pandas UDF](https://spark.apache.org/docs/3.4.2/api/python/reference/pyspark.sql/api/pyspark.sql.functions.pandas_udf.html) that contains the Torch code that will run on each worker to perform model inference. This inference job uses the [ResNet50](https://pytorch.org/vision/main/models/generated/torchvision.models.resnet50.html) model."
]
},
{

0 comments on commit 4139a99

Please sign in to comment.