auto-labeler / app /configs.py
dillonlaird's picture
initial commit
6723494
raw
history blame
No virus
115 Bytes
import torch
from pathlib import Path
DATA_ROOT = Path("data")
DEVICE = torch.device("cpu")
MODEL = "mobile_sam"