constants.py

PDFLATEX = "/external2/dkkoshman/venv/texlive/2022/bin/x86_64-linux/pdflatex"
GHOSTSCRIPT = "/external2/dkkoshman/venv/local/gs/bin/gs"

DATA_DIR = "data"
LATEX_PATH = "resources/latex.json"
TRAINER_DIR = "resources/trainer"
TOKENIZER_PATH = "resources/tokenizer.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