search

zubair-irshad / CenterSnap

Pytorch code for ICRA'22 paper: "Single-Shot Multi-Object 3D Shape Reconstruction and Categorical 6D Pose and Size Estimation"
https://zubair-irshad.github.io/projects/CenterSnap.html
286 stars 47 forks source link

about train #19

Open xuyb123 opened 1 year ago

xuyb123 commented 1 year ago

Hi, I had some problem in training:

  1. "trainer=pl.Trainer(..)" in net_train.py, the keyword "max_nb_epochs" "early_stop_callback" "checkpoint_callback" are not expected. image image

  2. " samples_per_epoch = len(train_ds.list())" in net_train.py will get 0? "train_ds" is the data_path.

zubair-irshad commented 1 year ago

  1. Thanks for your interest in our work! Can you do conda list and share your pytorch lightning and pytorch versions here. They should match the ones mentioned here

  2. Did you make sure your train_path and val path matches your downloaded data paths?