I am unable to reproduce the behavior that you are mentioning.

  
      gist.github.com
  
  
    https://gist.github.com/InnovArul/f5365f8986c0200e63c157ad06e794db 4
freeze_test.py
import torch, torch.nn as nn
import torch.nn.functional as F
import os, sys
import copy
import torch.optim as optim

class ConvReLU(nn.Module):
    def __init__(self, indim, outdim):
        super().__init__()
        self.conv = nn.Conv2d(indim, outdim, kernel_size=1)
This file has been truncated. show original