It is possible to use other models: medium, large, x..?

[mamanso]

Hi Xinze
I have created an python envirom to probe your code and it works fine with the examples. Crongratulations.

I am trying to train with my own data, 512x512 image tiles labeling bridges over roads, train, and rivers.

My question is related to how can i create a more complex model (medium, large, extra-large) to train it with my labeled data?
I can create de config file with de template for that models, and i can modify it but the starting weigths file, how can i create it? If a do no specify it download yolov5s.pt model and do no have expected information. Same when specify other models like medium or large.

Thank you