yingkaisha / keras-vision-transformer

The Tensorflow, Keras implementation of Swin-Transformer and Swin-UNET
MIT License
116 stars 40 forks source link

I run this code on my own dataset ,when I am saving the model (model .save) after 10 ,10 epoch I face this problem ,any body please help me #4

Open Nasiruddin696 opened 2 years ago

Nasiruddin696 commented 2 years ago

Capture

sciencehope commented 1 year ago

HI,

In my case, I have never able to save the model using model.save, instead, I use checkpoint to save model weight which is I think more convenient especially for custom model.