This is what my code looks like:
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") # Fields label_field = Field(se