search

yangxy / GPEN

2.44k stars 451 forks source link

fix OOM when loading pretrained weight, add starting iteration for re… #160

Open jinwonkim93 opened 2 years ago

jinwonkim93 commented 2 years ago

It raise Out of Memory Error when loading pretrained model in RTX 3090 x 4. I modified start_iter as argument for resuming training for issue below.

https://github.com/yangxy/GPEN/issues/82#issue-1082673510