Can I provide the weights of the trained model?

[ztx991028]

Can I provide the weights of the trained model?

[aofrancani]

Sure, I'd love to see that! My trained models are in https://github.com/aofrancani/TSformer-VO#2-pre-trained-models
Tks.

[ztx991028]

Whether you can provide a model that you have already trained, not a pre-trained model.

[aofrancani]

oh right, I sent the link with the weights of the trained models, and you could use them as pre-training. You just have to give the checkpoint path to load it properly.
Set the checkpoint path here: https://github.com/aofrancani/TSformer-VO/blob/main/train.py#L177
Also, see how I'm loading the checkpoint. If it is not "None", it will load the corresponding weights: https://github.com/aofrancani/TSformer-VO/blob/main/build_model.py#L82

[ztx991028]

great！