BatchSize set to 1 - Githubissues

limhoyeon / ToothGroupNetwork

3D Dental surface segmentation with Tooth Group Network

151 stars 38 forks source link

BatchSize set to 1 #23

Open XiShuFan opened 9 months ago

XiShuFan commented 9 months ago

Hello! I have a question for your inspiring work. Did you set batch size to 1 when training? Cause I find that you use BatchNorm in your network

limhoyeon commented 5 months ago

Yes I set batch size 1 which would be the same as layer normalization... Due to my limited gpu vram, i wanna set more batch size but i couldn`t do that..