Wei-JL

A strange bug encountered during training was the first propagation of the previous item in Variance_Attention.

2

``` TypeError: conv2d() received an invalid combination of arguments - got (Tensor, Parameter, NoneType, tuple, tuple, tuple, int), but expected one of: * (Tensor input, Tensor weight, Tensor bias, tuple...