Spaces:
Runtime error
Runtime error
File size: 471 Bytes
607ecc1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
import os
import gin
from .general import GeneralDataModule
@gin.configurable
class URMPDataModule(GeneralDataModule):
def __init__(
self,
urmp_root: str,
instrument: str,
batch_size: int = 16,
load_to_memory: bool = True,
**dataloader_args
):
super().__init__(
os.path.join(urmp_root, instrument),
batch_size,
load_to_memory,
**dataloader_args
)
|