This shape works on CUDA10.2 (PyTorch built from source) and CUDA10.1 (1.4.0 PyTorch binaries).
I get a “valid” result of “no faces were detected”. Do you get the same error running my code snippet?
import torch
import face_alignment
import numpy as np
fa = face_alignment.FaceAlignment(face_alignment.LandmarksType._2D, flip_input=False, device='cuda')
x = np.random.randn(224, 300, 3)
out = fa.get_landmarks(x)
> Warning: No faces were detected.