Training with another dataset?

DingXiaoH / RepVGG

RepVGG: Making VGG-style ConvNets Great Again

MIT License

3.3k stars 433 forks source link

Training with another dataset? #81

Closed Enescigdem closed 2 years ago

Enescigdem commented 2 years ago

Hello, Using https://github.com/rwightman/pytorch-image-models/ GitHub repo, I trained a model for my own data set. But when I tried to convert it to an inference time model.I got checkpoint loading inconsistencies. What should I do?

thanks in advance

DingXiaoH commented 2 years ago

You may load the checkpoint and convert, then save the converted model, so that you can build an inference-time model and load the saved checkpoint directly next time.