search

ranahanocka / MeshCNN

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

Specify the number of epochs #70

Closed claell closed 4 years ago

claell commented 4 years ago

Currently there does not seem to be an option to specify the number of epochs to use for training. I might look into adding that if I find the time.

ranahanocka commented 4 years ago

Hi @claell --

You can specify the # of training iterations by passing:

--niter 200 --niter_decay 1000 for example.

see here for more info

claell commented 4 years ago

Thanks for the hint! I looked through the options, but did not understand that niter means the number of epochs.

claell commented 4 years ago

Apparently I got that wrong. The number of epochs is actually the sum of niter and niter_decay.