search

intel-analytics / analytics-zoo

Distributed Tensorflow, Keras and PyTorch on Apache Spark/Flink & Ray
https://analytics-zoo.readthedocs.io/
Apache License 2.0
11 stars 3 forks source link

Use saveModule to save checkpoint instead of save in optimizer. #37

Open qiuxin2012 opened 2 years ago

qiuxin2012 commented 2 years ago

Due to our user's feedback, his model is too big, and java serialization doesn't work.