Yaoyi-Li / GCA-Matting

Official repository for Natural Image Matting via Guided Contextual Attention
MIT License
390 stars 62 forks source link

Why is resuming training is uncomment? #17

Open Feywell opened 3 years ago

Feywell commented 3 years ago

I try to train a model by my own dataset. But it is interrupt. How can I resume trainging

It will return a error when using resume training:

RuntimeError: Caught RuntimeError in replica 0 on device 0.

Yaoyi-Li commented 3 years ago

You can try to uncomment this line to resume the training which is interrupted. https://github.com/Yaoyi-Li/GCA-Matting/blob/5a999be4b00bb482658fa7013a74a2b8acddc78e/config/gca-dist.toml#L21

ucasiggcas commented 3 years ago

hi,dear when you resume training with your dataset how to set the config file ? could you please show me the file ? thx https://github.com/Yaoyi-Li/GCA-Matting/issues/25