[COLOR=var(--ansi-red)]TypeError[/COLOR] Traceback (most recent call last)
<ipython-input-33-505edb0b4bd3> in [COLOR=var(--ansi-cyan)]<module>[/COLOR]
[COLOR=var(--ansi-green)]
14[/COLOR] optimizer[COLOR=var(--ansi-blue)].[/COLOR]zero_grad[COLOR=var(--ansi-blue)]([/COLOR][COLOR=var(--ansi-blue)])[/COLOR]
[COLOR=var(--ansi-green)]
15[/COLOR] [COLOR=var(--ansi-red)]# Forward pass[/COLOR]
[COLOR=var(--ansi-green)]---> 16[/COLOR][COLOR=var(--ansi-red)] [/COLOR]output [COLOR=var(--ansi-blue)]=[/COLOR] model[COLOR=var(--ansi-blue)]([/COLOR]data[COLOR=var(--ansi-blue)])[/COLOR]
[COLOR=var(--ansi-green)]
17[/COLOR] [COLOR=var(--ansi-red)]# Calculating loss[/COLOR]
[COLOR=var(--ansi-green)]
18[/COLOR] loss [COLOR=var(--ansi-blue)]=[/COLOR] loss_func[COLOR=var(--ansi-blue)]([/COLOR]output[COLOR=var(--ansi-blue)],[/COLOR] target[COLOR=var(--ansi-blue)])[/COLOR]
/usr/local/lib/python3.7/dist-packages/torch/nn/modules/conv.py in [COLOR=var(--ansi-cyan)]_conv_forward[/COLOR][COLOR=var(--ansi-blue)](self, input, weight, bias)[/COLOR]
[COLOR=var(--ansi-green)]
452[/COLOR] _pair(0), self.dilation, self.groups)
[COLOR=var(--ansi-green)]
453[/COLOR] return F.conv2d(input, weight, bias, self.stride,
[COLOR=var(--ansi-green)]--> 454[/COLOR][COLOR=var(--ansi-red)] self.padding, self.dilation, self.groups)
[/COLOR][COLOR=var(--ansi-green)]
455[/COLOR]
[COLOR=var(--ansi-green)]
456[/COLOR] [COLOR=var(--ansi-green)]def[/COLOR] forward[COLOR=var(--ansi-blue)]([/COLOR]self[COLOR=var(--ansi-blue)],[/COLOR] input[COLOR=var(--ansi-blue)]:[/COLOR] Tensor[COLOR=var(--ansi-blue)])[/COLOR] [COLOR=var(--ansi-blue)]->[/COLOR] Tensor[COLOR=var(--ansi-blue)]:[/COLOR]
[COLOR=var(--ansi-red)]TypeError[/COLOR]: conv2d() received an invalid combination of arguments - got (tensorflow.python.framework.ops.EagerTensor, Parameter, Parameter, tuple, tuple, tuple, int), but expected one of:
* (Tensor input, Tensor weight, Tensor bias, tuple of ints stride, tuple of ints padding, tuple of ints dilation, int groups)
didn't match because some of the arguments have invalid types: (!tensorflow.python.framework.ops.EagerTensor!, !Parameter!, !Parameter!, !tuple!, !tuple!, !tuple!, int)
* (Tensor input, Tensor weight, Tensor bias, tuple of ints stride, str padding, tuple of ints dilation, int groups)
didn't match because some of the arguments have invalid types: (!tensorflow.python.framework.ops.EagerTensor!, !Parameter!, !Parameter!, !tuple!, !tuple!, !tuple!, int)