new(): invalid data type ‘str‘

new(): invalid data type 'str'

from collections import OrderedDict

import torch

from LPRNetN.model.STN import STNet

STN = STNet()

dummy_input1 = torch.randn(1, 3, 24, 94)



state_dict1 = torch.load('weights/STNNet.pth', map_location=lambda storage, loc: storage)
state_dict2 = torch.load('weights/STNNet.pth', map_location=lambda storage, loc: storage)
new_state_dict1 = OrderedDict()
new_state_dict2 = OrderedDict()
for k, v in state_dict2.items():
    name = k.replace('module.', '')  # remove `module.`
    new_state_dict2[name] = v

for k, v in state_dict1.items():
    name = k.replace('module.', '')  # remove `modul