FaceX-Zoo icon indicating copy to clipboard operation
FaceX-Zoo copied to clipboard

a little bug found

Open yutaoxxx opened this issue 2 years ago • 1 comments

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()

yutaoxxx avatar Jun 20 '22 02:06 yutaoxxx

Thanks!

wang21jun avatar Jun 20 '22 02:06 wang21jun