EcoDepth icon indicating copy to clipboard operation
EcoDepth copied to clipboard

Training reproduction problems

Open wanshishuns opened this issue 7 months ago • 2 comments

I'm very sorry to disturb you again. I encountered a problem when reproducing your code. I commented on the following code in train.py:

if args.ckpt_path == "":

model_str = f"weights_{args.scene}.ckpt"

download_model(model_str)

args.ckpt_path = f"../checkpoints/{model_str}"

model.load_state_dict(torch.load(args.ckpt_path, map_location="cpu", weights_only=True)["state_dict"])

because I need to train from scratch and this section needs to be commented on. I don't know if this is correct. At the same time, I commented on the code in model.py.

Only support finetuning for now

    # assert not args.train_from_scratch

After I completed the training,The metrics are very low and the reproduction is not well achieved. I haven't modified any other code configurations. d1 only have 0.5 d2 only have 0.6 and d3 only have 0.9 Do you know the reason? I conducted the training on RTX3090, with batch set to 2 and num_workers to 8. However, the influence of batch should not be so huge. I wonder if you know the reason. And are there any problems with my operation?

I look forward to your answer. Thank you very much for your help to me! Thank you very much!

wanshishuns avatar May 14 '25 03:05 wanshishuns

Please take a look at https://github.com/Aradhye2002/EcoDepth/issues/25.

Aradhye2002 avatar May 14 '25 05:05 Aradhye2002

thank you!

wanshishuns avatar May 14 '25 07:05 wanshishuns