cuda running out of memory with vox 256x256 dataset chekpoint. suggest edits pls.

AliaksandrSiarohin / monkey-net

Animating Arbitrary Objects via Deep Motion Transfer

467 stars 81 forks source link

Open kashi211 opened 4 years ago

kashi211 commented 4 years ago

cuda running out of memory with vox 256x256 dataset chekpoint. suggest edits pls.

AliaksandrSiarohin commented 4 years ago

How much gpu memore you have? Try to run on cpu, by removing all calls to cuda.