'Tensor' object has no attribute 'uint8'

[tx871217]

Hi,

I was running the "test.py' with python test.py --name xxx --netG local --resize_or_crop none --data_type 8 --dataroot './datasets/xxxx' --loadSize 256 --fineSize 256 --label_nc 0 --no_instance --ngf 32 and got error AttributeError: 'Tensor' object has no attribute 'uint8' at line 45 Was this because the Pytorch version? Mine is 1.9.0 Help me solve this, plzzz~

[flyingmrwang]

The same for torch 1.9 After change *.uint8() to *.to(torch.uint8), this part passed. However face another error:

[flyingmrwang]

after wrapping export with torch.no_grad(), previous one passed.. While here comes another one:

[MikhailKulyabin]

Hi, did someone solve it?