I got this error while trying to convert nn.module to torchscript.
RuntimeError:
undefined value super:
  File "D:\project\Kikai project\DIFRINT\models\correlation\correlation.py", line 147
        def __init__(self):
                super(FunctionCorrelation, self).__init__()
  ~~~~~ <--- HERE
'FunctionCorrelation.__init__' is being compiled since it was called from '__torch__.models.correlation.correlation.FunctionCorrelation'
  File "D:\project\Kikai project\DIFRINT\models\correlation\correlation.py", line 238
        def forward(self, tensorFirst, tensorSecond):
                return FunctionCorrelation.apply(tensorFirst, tensorSecond)
         ~~~~~~~~~~~~~~~~~~~ <--- HERE
'__torch__.models.correlation.correlation.FunctionCorrelation' is being compiled since it was called from 'ModuleCorrelation.forward'
  File "D:\project\Kikai project\DIFRINT\models\correlation\correlation.py", line 238
        def forward(self, tensorFirst, tensorSecond):
                return FunctionCorrelation.apply(tensorFirst, tensorSecond)
         ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ <--- HERE

This is my code
class FunctionCorrelation(torch.autograd.Function):
	def __init__(self):
		super(FunctionCorrelation, self).__init__()
		# super().__init__()
	# end

	@staticmethod
	def forward(self, first, second):
		self.save_for_backward(first, second)

and
class ModuleCorrelation(torch.nn.Module):
	def __init__(self):
		super(ModuleCorrelation, self).__init__()
		# super().__init__()
	# end

	def forward(self, tensorFirst, tensorSecond):
		return FunctionCorrelation.apply(tensorFirst, tensorSecond)
	# end

I used torch==1.6.0, torchvision==0.7.0,  cupy-cuda102==7.8.0, Pillow==6.1.0
I think may be get problem when i use torch.autograd.Function  this cause i have to use @staticmethod in torch==1.6.0
Someone can help me
thank