ShunChengWu / 3DSSG

Other
130 stars 20 forks source link

Question about Training SGFN #60

Closed J-C-Y closed 1 week ago

J-C-Y commented 3 weeks ago

Hi Thanks for your great work!

I have some questions about training SGFN using 3RSCAN dataset.

I run the train with the code that you provided.

python main.py --mode train --config [PATH to config_SGFN_full_l160]

But the loss did not decrease and remained around 5.

I checked that loss_rel was reduced to around 0.0x, but the loss_obj remained around 5.

What's the problem? Can you check the code?

I would appreciate it if you could share how the loss fell.

image

image

image

ShunChengWu commented 3 weeks ago

Hm.. there are many possibilities. It's possible that the dataset is wrong or any other settings are incorrect. It's also possible it's the initialization problem of the model weight. Can you try to run the pre-trained model to see if your setup are correct?