TextAttack
TextAttack copied to clipboard
Train model on custom dataset from file does not work
Describe the bug Train model on custom dataset returns failure.
To Reproduce Steps to reproduce the behavior:
- Run following command
!textattack train --model distilbert-base-uncased --dataset-from-file <*.py> --model-num-labels 2 --model-max-length 64 --per-device-train-batch-size 128 --num-epochs 3
Expected behavior Expected to train the model from the custom dataset
Screenshots or Traceback
System Information (please complete the following information):
- OS: Windows
- Textattack version: latest
@DavidHerel I don't think the library currently supports training using --dataset-from-file , only attack that's why you are hitting the error.
@jxmorris12 Can you confirm that we currently do not support training using dataset from file ? If so , should we mark this as an enhancement ?