batch_size: 128
card_output_dim: 512
dropout: 0.3
embedding_size: 4130
imitation_learning: true
input_size: 4130
lambda_gih: 1.0
lr: 0.0001
max_choices: 15
max_epochs: 100
net_name: network
num_card_layers: 15
num_workers: 16
path: null
persistent_workers: true
pin_memory: true
prefetch_factor: 16
raw_data_folder: data/raw_data
tau: 0.7
test_sets:
- BRO
train_sets:
- SOS
warmup_epochs: 3