File size: 391 Bytes
6fc49ec e3211d6 6fc49ec | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 | 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
|