dmmiller612 / sparktorch

Train and run Pytorch models on Apache Spark.
MIT License
339 stars 44 forks source link

Implement mini-batches for training to reduce CPU burden for large datasets in partitions. #4

Closed dmmiller612 closed 4 years ago