search

JuliaWolleb / Diffusion-based-Segmentation

This is the official Pytorch implementation of the paper "Diffusion Models for Implicit Image Segmentation Ensembles".
MIT License
272 stars 35 forks source link

I meet a memory question when i run this code? #14

Closed Wanyidon closed 1 year ago

Wanyidon commented 1 year ago

I set the same parameters as you except the batch_size. i have set the batch size =1, however, the system still show CUDA error out of memory. My GPU is 1080ti

JuliaWolleb commented 1 year ago

this is weird, with a batch-size of 1 this should work. do you still have this error?

MuqinZ commented 1 year ago

Can you give me your E-mail. I want to talk about this Net-Model with you.