Solved by Valerio_Biscione in post #10 
              
              
                Here you go 
import torch
from torch import nn
import csv
import copy


class Net(nn.Module):

    def __init__(self, old_state_dict, state_dict_map):

        # run nn.Module's constructor

        super(Net, self).__init__()

        # --------------------------------------------------------------…