timoschick / pet

This repository contains the code for "Exploiting Cloze Questions for Few-Shot Text Classification and Natural Language Inference"
https://arxiv.org/abs/2001.07676
Apache License 2.0
1.62k stars 283 forks source link

PET and iPET parameters #88

Open KsLimon opened 2 years ago

KsLimon commented 2 years ago

Is it possible to train ipet with pet parameters which are prefixed with --pet_? As a sample,

python cli.py \
--method ipet \
--pattern_ids 0 1 2 \
--data_dir /- \
--model_type bert \
--model_name_or_path /- \
--task_name - \
--output_dir agnews-output \
--do_train \
--do_eval \
--train_examples 189186 \
--unlabeled_examples 5802 \
--split_examples_evenly \
--pet_per_gpu_train_batch_size 1 \
--pet_per_gpu_unlabeled_batch_size 3 \
--pet_gradient_accumulation_steps 4 \
--pet_max_steps 250 \