Closed zwishenzug closed 1 year ago
Some models don't have a state dict, they just skip that layer and have the keys directly.
After the torch.load parts I had to add some code like this to wrap those models:
if not 'state_dict' in model_a: model_a = {'state_dict': model_a}
this saved my butt, thanks!
Some models don't have a state dict, they just skip that layer and have the keys directly.
After the torch.load parts I had to add some code like this to wrap those models: