running in 3D produces error that G is undefined on line 302 (network3D.py)

[aziabari]

When I run the code in 3D, I get an error that says G is undefined on line 302 (network3D.py)

I believe the following lines

296: self.F = self.build_conv_block(dim // 2, True, norm_layer) 297: self.G = self.build_conv_block(dim // 2, True, norm_layer)

should be replaced by:

296: F = self.build_conv_block(dim // 2, True, norm_layer) 297: G = self.build_conv_block(dim // 2, True, norm_layer)

Is this the correct solution?