Skip to content

GitLab

Unverified Commit 7d40b2ac authored by moneypi's avatar moneypi Committed by GitHub
Browse files

fix ValueError for adversarial_text (#9163)

parent 09050ac3
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
research/adversarial_text/README.md
View file @ 7d40b2ac
...@@ -54,7 +54,7 @@ $ PRETRAIN_DIR=/tmp/models/imdb_pretrain ...@@ -54,7 +54,7 @@ $ PRETRAIN_DIR=/tmp/models/imdb_pretrain
$ python pretrain.py \ $ python pretrain.py \
--train_dir=$PRETRAIN_DIR \ --train_dir=$PRETRAIN_DIR \
--data_dir=$IMDB_DATA_DIR \ --data_dir=$IMDB_DATA_DIR \
--vocab_size=86934 \ --vocab_size=87007 \
--embedding_dims=256 \ --embedding_dims=256 \
--rnn_cell_size=1024 \ --rnn_cell_size=1024 \
--num_candidate_samples=1024 \ --num_candidate_samples=1024 \
...@@ -83,7 +83,7 @@ $ python train_classifier.py \ ...@@ -83,7 +83,7 @@ $ python train_classifier.py \
--train_dir=$TRAIN_DIR \ --train_dir=$TRAIN_DIR \
--pretrained_model_dir=$PRETRAIN_DIR \ --pretrained_model_dir=$PRETRAIN_DIR \
--data_dir=$IMDB_DATA_DIR \ --data_dir=$IMDB_DATA_DIR \
--vocab_size=86934 \ --vocab_size=87007 \
--embedding_dims=256 \ --embedding_dims=256 \
--rnn_cell_size=1024 \ --rnn_cell_size=1024 \
--cl_num_layers=1 \ --cl_num_layers=1 \
...@@ -111,7 +111,7 @@ $ python evaluate.py \ ...@@ -111,7 +111,7 @@ $ python evaluate.py \
--run_once \ --run_once \
--num_examples=25000 \ --num_examples=25000 \
--data_dir=$IMDB_DATA_DIR \ --data_dir=$IMDB_DATA_DIR \
--vocab_size=86934 \ --vocab_size=87007 \
--embedding_dims=256 \ --embedding_dims=256 \
--rnn_cell_size=1024 \ --rnn_cell_size=1024 \
--batch_size=256 \ --batch_size=256 \
......
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