13 lines
293 B
Python
13 lines
293 B
Python
from pathlib import Path
|
|
import torch
|
|
|
|
BATCH_SIZE = 128
|
|
NUM_EPOCHS = 50
|
|
NUM_WORKERS = 4
|
|
NUM_SEGMENTS = 5
|
|
NUM_SEGMENT_HOPS = 2
|
|
SEEDS = [42, 1337]
|
|
ALL_DATASET_PATHS = list((Path(__file__).parent.absolute() / 'data' / 'mimii').glob('*/'))
|
|
|
|
DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
|