File size: 667 Bytes
61e1114 |
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 26 |
from pathlib import Path
def get_config():
return {
"batch_size":2,
"num_epochs": 100,
"lr": 10**-4,
"seq_len": 150,
"d_model": 512,
"lang_src": "0",
"lang_tgt": "1",
"model_folder": "weights",
"model_basename": "tmodel_",
"preload": None,
"tokenizer_file": "tokenizer_{0}.json",
"experiment_name": "runs/tmodel"
}
def get_weights_file_path(config, epoch: str):
model_folder = config["model_folder"]
model_basename = config["model_basename"]
model_filename = f"{model_basename}{epoch}.pt"
return str(Path('.') / model_folder / model_filename)
|