configs/vqa_med_2019/c1_classification/c1_classification_question_rnn.yml
# Load config defining tasks for training, validation and testing.
default_configs: vqa_med_2019/c1_classification/default_c1_classification.yml
pipeline:
# Questions encoding.
question_tokenizer:
type: SentenceTokenizer
priority: 1.1
streams:
inputs: questions
outputs: tokenized_questions
# Model 1: Embeddings
question_embeddings:
type: SentenceEmbeddings
priority: 1.2
embeddings_size: 50
pretrained_embeddings_file: glove.6B.50d.txt
data_folder: ~/data/vqa-med
word_mappings_file: questions.all.word.mappings.csv
streams:
inputs: tokenized_questions
outputs: embedded_questions
# Model 2: RNN
lstm:
type: RecurrentNeuralNetwork
cell_type: LSTM
prediction_mode: Last
priority: 3
initial_state: Zero
#num_layers: 5
hidden_size: 50
streams:
inputs: embedded_questions
globals:
input_size: embeddings_size
prediction_size: vocabulary_size_c1
#: pipeline