Issue related to Training the model

[Aryan-Koul29]

Sir, we have successfully started the training of the model. However, at the end of the training, we are getting files named model.gen.pth and model.dis.pth. When we attempt to load this model in the demo_gannotation.py file, it gives us an error stating that it cannot read the model. Additionally, the model you have provided in the repository only has the .pth extension. What steps should we take to obtain a similar model?

Also Sir can you please share the pts.pt file of any of the training dataset of yours. If you could only tell the structure of the file, it is also fine.