Freeze Layers

for params in net.parameters():
  params.require_grad = False