nv-tlabs / GET3D

Other
4.2k stars 376 forks source link

Till what kimg value should i train this model #90

Closed dheeraj-pant closed 1 year ago

dheeraj-pant commented 1 year ago

Hi, I am training Get3d on motorbike shapenet images.

Input data description: 337 motorbike -----> Each with 24 images and camera matrix

Training Command: python train_3d.py --outdir=PATH_TO_LOG --data=PATH_TO_RENDER_IMG --camera_path PATH_TO_RENDER_CAMERA --gpus=8 --batch=32 --gamma=80 --data_camera_mode shapenet_motorbike --dmtet_scale 1.0 --use_shapenet_split 1 --one_3d_generator 0

I check the model pt files for inference and the result is not like your paper not a good geometry one created.

I see by default its training for 20k kimg. Which value should i pick for this as its not mention in paper .

SteveJunGao commented 1 year ago

Hi @dheeraj-pant,

The default is the maximum iterations we used for training, but typically, we choose the checkpoint that achieves the best FID on the validation set. You can use the tensorboard to visualize the FID on the validation set (we logged it in this line)

samrat213 commented 1 year ago

you would start to see results around 100kimgs. if not something went wrong with your changes.

SteveJunGao commented 1 year ago

@dheeraj-pant, I'm closing this issue since there has been no update for 3 weeks. Feel free to reopen it if you have any questions!