import torch

DEVICE = "cuda" if torch.cuda.is_available() else "cpu"

# IMAGE_DIR = "<TRAIN IMAGE DIRECTORY HERE>"
IMAGE_DIR = "EnglishLP"
ANNOTATION_CSV = "dataset_lp_labels.csv"
ANCHOR_DIR = "lp_anchor.txt"
# FOLD_FILE = "<FOLD FILE PATH HERE>"
FOLD_FILE = "EnglishLP_folds.json"
CURRENT_FOLD = "fold1"
# RUNTIME_INFO_FILE = "drive/MyDrive/LP_Detection/runtime_info.json"
# CHECKPOINT_FILE = "drive/MyDrive/LP_Detection/checkpoint.pth.tar"
RUNTIME_INFO_FILE = "runtime_info.json"
CHECKPOINT_FILE = "checkpoint.pth.tar"

ORIGINAL_SIZE = 416
IMAGE_WIDTH = 416
IMAGE_HEIGHT = 416
NUM_CLASSES = 3

BATCH_SIZE = 64
SUBDIVISIONS = 8
LEARNING_RATE = 1e-4
STEPS = [100, 72000, 81000]
MOMENTUM = 0.9
WEIGHT_DECAY = 5e-4
MAX_BATCH = 90000

CONF_THRESHOLD = 0.01

PIN_MEMORY = True
LOAD_MODEL = True
SAVE_MODEL = True

import torch

DEVICE = "cuda" if torch.cuda.is_available() else "cpu"

# IMAGE_DIR = "<TRAIN IMAGE DIRECTORY HERE>"
IMAGE_DIR = "EnglishLP"
ANNOTATION_CSV = "dataset_lp_labels.csv"
ANCHOR_DIR = "lp_anchor.txt"
# FOLD_FILE = "<FOLD FILE PATH HERE>"
FOLD_FILE = "EnglishLP_folds.json"
CURRENT_FOLD = "fold1"
# RUNTIME_INFO_FILE = "drive/MyDrive/LP_Detection/runtime_info.json"
# CHECKPOINT_FILE = "drive/MyDrive/LP_Detection/checkpoint.pth.tar"
RUNTIME_INFO_FILE = "runtime_info.json"
CHECKPOINT_FILE = "checkpoint.pth.tar"

ORIGINAL_SIZE = 416
IMAGE_WIDTH = 416
IMAGE_HEIGHT = 416
NUM_CLASSES = 3

BATCH_SIZE = 64
SUBDIVISIONS = 8
LEARNING_RATE = 1e-4
STEPS = [100, 72000, 81000]
MOMENTUM = 0.9
WEIGHT_DECAY = 5e-4
MAX_BATCH = 90000

CONF_THRESHOLD = 0.01

PIN_MEMORY = True
LOAD_MODEL = True
SAVE_MODEL = True

# IMAGE_DIR = "<TRAIN IMAGE DIRECTORY HERE>" # FOLD_FILE = "<FOLD FILE PATH HERE>" # RUNTIME_INFO_FILE = "drive/MyDrive/LP_Detection/runtime_info.json" # CHECKPOINT_FILE = "drive/MyDrive/LP_Detection/checkpoint.pth.tar"