import torch
model = torch.load("models/eca.pth.tar")

model.keys()

torch.save(model['state_dict'], 'models/eca_net.pth.tar')

modelx = torch.load("models/eca_net.pth.tar")

modelx.keys()