pere commited on
Commit
bdaac68
1 Parent(s): 9f5bf65

first push

Browse files
Files changed (1) hide show
  1. run_w7.sh +2 -1
run_w7.sh CHANGED
@@ -3,7 +3,8 @@
3
  --model_type="roberta" \
4
  --config_name="./" \
5
  --tokenizer_name="./" \
6
- --dataset_name="NbAiLab/NCC_small" \
 
7
  --max_seq_length="128" \
8
  --weight_decay="0.01" \
9
  --per_device_train_batch_size="128" \
 
3
  --model_type="roberta" \
4
  --config_name="./" \
5
  --tokenizer_name="./" \
6
+ --train_file="/mnt/disks/flaxdisk/corpus/train-shard-0001-of-0001.json" \
7
+ --validation_file="/mnt/disks/flaxdisk/corpus/validation-shard-0001-of-0001.json" \
8
  --max_seq_length="128" \
9
  --weight_decay="0.01" \
10
  --per_device_train_batch_size="128" \