File size: 540 Bytes
c308f77
 
 
96feb73
 
 
 
 
 
 
 
c308f77
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
PDFLATEX = "/external2/dkkoshman/venv/texlive/2022/bin/x86_64-linux/pdflatex"
GHOSTSCRIPT = "/external2/dkkoshman/venv/local/gs/bin/gs"

DATA_DIR = "local/data"
WANDB_DIR = "local/wandb"
TRAINER_DIR = "local/trainer"

RESOURCES = "resources"
LATEX_PATH = RESOURCES + "/latex.json"
TOKENIZER_PATH = RESOURCES + "/tokenizer.pt"
DATAMODULE_PATH = RESOURCES + "/datamodule.pt"

NUM_DATALOADER_WORKERS = 4
PERSISTENT_WORKERS = True  # whether to shut down workers at the end of epoch
PIN_MEMORY = False  # probably causes cuda oom error if True