auto-labeler / app /configs.py
dillonlaird's picture
attempting to fix permission issue on creating data.pkl, part 3
702f0a9
import os
import torch
from pathlib import Path
DATA_ROOT = Path(os.environ["HOME"]) / Path("app") / Path("data")
DEVICE = torch.device("cpu")
MODEL = "mobile_sam"