Spaces:
Runtime error
Runtime error
File size: 962 Bytes
68c2d97 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
from torch.utils.data import Dataset
import numpy as np
import torch
import os
import pickle
# from utils import *
# from constants import get_soft_mask, get_soft_mask2, rand_sample_mask
class baseDataset(Dataset):
def __init__(self, folders, logger, seq_len, use_cfg, cfg_p, scale):
self.folders = folders
self.logger = logger
self.seq_len = seq_len
self.use_cfg = use_cfg
self.cfg_p = cfg_p
self.scale = scale
self.data = {}
self.len = 0
def load_data(self, folder):
pass
def __len__(self):
return self.len
def __getitem__(self, idx):
pass
def print_config(self, **kwargs):
print("In dataset")
print("Folders: ", self.folders)
print("Seq len: ", self.seq_len)
print("Use cfg: ", self.use_cfg)
print("Cfg p: ", self.cfg_p)
print("Scale: ", self.scale)
print("Len: ", self.len)
|