And when i change the network to ResNet in get_net i am getting error
INPUT = 'noise' # 'meshgrid'           get_noise function
pad = 'reflection'
OPT_OVER = 'net' # 'net,input'

reg_noise_std = 1./30. # set to 1./20. for sigma=50
LR = 0.01

OPTIMIZER='adam' # 'LBFGS'
show_every = 100
exp_weight=0.99


num_iter = 1000
input_depth = 3 
figsize = 4 
    
    
net = get_net(input_depth, 'ResNet', pad, upsample_mode='bilinear').type(dtype)


    
net_input = get_noise(input_depth, INPUT, (img_pil.size[1], img_pil.size[0])).type(dtype).detach()

# Compute number of parameters
s  = sum([np.prod(list(p.size())) for p in net.parameters()]); 
print ('Number of params: %d' % s)

# Loss
mse = torch.nn.MSELoss().type(dtype)

img_noisy_torch = np_to_torch(img_noisy_np).type(dtype)

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-7-7c96e1404ffa> in <module>()
     16 
     17 
---> 18 net = get_net(input_depth, 'ResNet', pad, upsample_mode='bilinear').type(dtype)
     19 
     20 

2 frames
/content/models/common.py in act(act_fun)
     90             assert False
     91     else:
---> 92         return act_fun()
     93 
     94 

TypeError: __init__() missing 1 required positional argument: 'num_features'

@ptrblck