File size: 306 Bytes
464ed03 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
def getConfig():
model_config = {
"emsize" : 300,
"d_hid" : 1024,
"nlayers" : 6,
"nhead" : 6,
"dropout" : 0.2,
"bptt" : 64
}
app_config = {
"logs" : "tensorboard_logs",
"epochs" : 25,
}
return model_config, app_config
|