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