Hello, I want to test the weights I trained myself, but I can't find the folder where the weights are loaded.

[Inverse-function]

Hello, I want to test the weights I trained myself, but I can't find the folder where the weights are loaded.

[siyandong]

Hi, thank you for your interest in our work. In our test scripts, we use the weights hosted on HuggingFace, which correspond to the model we trained during paper submission. You can simply load your checkpoints as standard PyTorch code, e.g.,

reloc3r/train.py

Line 154 in 62ad49e

print(model.load_state_dict(ckpt, strict=False))

If you are using our training scripts, your trained weights will be saved to the folder ./checkpoints/xxx