search

ranahanocka / MeshCNN

Convolutional Neural Network for 3D meshes in PyTorch
MIT License
1.56k stars 314 forks source link

Meaning of num_groups #65

Closed claell closed 4 years ago

claell commented 4 years ago

I don't know what the num_groups parameter does. At first I thought it is the number of classes used for training, but apparently that is not correct.

ranahanocka commented 4 years ago

It is a hyper-parameter for the groupnorm normalization layer. You can see more info in the PyTorch docs for GroupNorm.

claell commented 4 years ago

Thanks for the explanation. I should have looked into the options file here.