I had a bigger dataset and i trained my model with 5 folds in it , each fold will give a best weights file. Now is there any way to merge all the 5 weights file into single weight file.
My Model:
class ImgClassifier(nn.Module):
def __init__(self, model_arch, n_class, pretrained=False):
super().__init__()
self.model = timm.create_model(model_arch, pretrained=pretrained)
n_features = self.model.classifier.in_features
self.model.classifier = nn.Linear(n_features, n_class)
def forward(self, x):
x = self.model(x)
return x
how to acheive it ? any idea would be helpful .
TIA