Open rexayyy opened 2 years ago
i have trained a model in pytorch script using yolov6 in colab. I have the weight files with me. How can i load my model trained on custom data.
Same answer.
You can set the pretrained model path here:
No, i am asking you about the a custom trained model not a pretrained model. like you load one using model.load command
You can set the pretrained model path here: https://github.com/meituan/YOLOv6/blob/ab76a39b9f89c15424ad6366794306239caaf573/configs/yolov6s.py#L4