alverciito
upload safetensors and refactor research files
dbd79bd
raw
history blame contribute delete
984 Bytes
# - x - x - x - x - x - x - x - x - x - x - x - x - x - x - #
# #
# This file was created by: Alberto Palomo Alonso #
# Universidad de Alcalá - Escuela Politécnica Superior #
# #
# - x - x - x - x - x - x - x - x - x - x - x - x - x - x - #
# Import statements:
from dataclasses import dataclass
@dataclass
class DatasetConfig:
# Paths:
train_data_path: str = None
val_data_path: str = None
test_data_path: str = None
# Percentages:
train_percentage: float = 1.0
val_percentage: float = 1.0
test_percentage: float = 1.0
# Other parameters:
num_workers: int = 0
shuffle_train: bool = True
shuffle_val: bool = True
# - x - x - x - x - x - x - x - x - x - x - x - x - x - x - #
# END OF FILE #
# - x - x - x - x - x - x - x - x - x - x - x - x - x - x - #