FaceX-Zoo
FaceX-Zoo copied to clipboard
a little bug found
addition_module/face_mask_adding/FMA_3D/face_masker.py
line 58 as follow.
if torch.cuda.is_available:
cropped_image = cropped_image.cuda()
should be if torch.cuad.is_available()
Thanks!