mtg-draft-model / config.yaml
TimoBertram's picture
Upload config.yaml with huggingface_hub
e3211d6 verified
Raw
History Blame Contribute Delete
391 Bytes
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