Commit 985bed5a authored by Gaëtan Caillaut's avatar Gaëtan Caillaut
Browse files

suppression du device passé à HateSpeechCollater

parent d426d9a2
......@@ -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):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment