JuliaWolleb / Diffusion-based-Segmentation

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

Can you tell me whether Nvidia 3090 with 24GB is able to train model? #66

Open Moriatyyuan opened 7 months ago

Moriatyyuan commented 7 months ago

CleanShot 2024-04-07 at 23 55 10@2x

sherrlockk commented 6 months ago

yes,check to any other workouts going on

princerice commented 6 months ago

I want to know the size of the batch you set up

sherrlockk commented 6 months ago

According to this paper, 10 batch_size, 3090 is perfectly sufficient

Moriatyyuan commented 6 months ago

According to this paper, 10 batch_size, 3090 is perfectly sufficient

thank you, I will try later