search

JDAI-CV / FaceX-Zoo

A PyTorch Toolbox for Face Recognition
https://arxiv.org/pdf/2101.04407.pdf
Other
1.87k stars 434 forks source link

a little bug found #150

Open yutaoxxx opened 2 years ago

yutaoxxx commented 2 years ago

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

wang21jun commented 2 years ago

Thanks!