Hi @Shisho_Sama ,
I am also having the same error.
I used the same github repo.
What changes did you make in the code to make it up and running
?
I used the below code.
import torch
import torchvision
import face_alignment

Optionally set detector and some additional detector parameters
face_detector = ‘sfd’
face_detector_kwargs = {
“filter_threshold” : 0.8
}
model = face_alignment.FaceAlignment(face_alignment.LandmarksType._3D, device=‘cpu’, flip_input=True,face_detector=face_detector)
example = io.imread(’/test/assets/aflw-test.jpg’)
traced_script_module = torch.jit.trace(model,example)
traced_script_module.save(“traced_facealignment_model.pt”)