I get the error
AssertionError: The input data type needs to be fp16 but torch.float32 was found!
· Sign up or log in to comment