g_AtoB = Generator().type(dtype) UnboundLocalError: local variable 'dtype' referenced before assignment please help