suppression du device passé à HateSpeechCollater

train_roberta.py

@@ -19,7 +19,7 @@ if __name__ == "__main__":
 
         tokenizer = RobertaTokenizerFast.from_pretrained('roberta-base')
         dataset = HateSpeechDataset(args.input)
-        collater = HateSpeechCollater(tokenizer, device)
+        collater = HateSpeechCollater(tokenizer)
         class_names = collater.class_names()
 
         for fold, (train, test) in enumerate(dataset.iter_folds(args.folds, True), 1):