niuchuangnn / SPICE

Other
208 stars 47 forks source link

SPICE on single GPU #35

Open TomasPlachy opened 2 years ago

TomasPlachy commented 2 years ago

It seems like you have prepared arguments to run the script on single GPU (--multiprocessing-distributed and --gpu), but then I run into "Only DistributedDataParallel is supported." exception. Or when I run the script only with "--gpu 0" as argument, I run into error originating in spawn .py code.

I only have access to a computer with single Nvidia 3090 GPU and want to test your model as part of my Master's thesis. Does it mean that I would need to implement a version for 1 GPU myself, or am I missing something in your implementation?

Adonis-Lbc commented 1 year ago

Hello I have a similar problem, can you solve it? Thanks

myymgkm123 commented 8 months ago

Hello, me too, can you solve it? Thanks