Spaces:
Running
Running
File size: 382 Bytes
08a644f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
import os
from pathlib import Path
HF_TOKEN = os.environ.get("HF_TOKEN")
ROOT_EXP_DIR = Path(
os.environ.get("ROOT_EXP_DIR")
or os.path.join(os.path.dirname(os.path.abspath(__file__)), "../logs")
).resolve()
ROOT_EXP_DIR.mkdir(exist_ok=True, parents=True)
BATCH_SIZE = int(os.environ.get("BATCH_SIZE") or 8)
TRAINING_EPOCHS = int(os.environ.get("TRAINING_EPOCHS") or 10)
|