from spirl.utils.general_utils import AttrDict from spirl.components.data_loader import GlobalSplitVideoDataset data_spec = AttrDict( dataset_class=GlobalSplitVideoDataset, n_actions=3, state_dim=23, split=AttrDict(train=0.95, val=0.05, test=0.0), res=32, crop_rand_subseq=True, ) data_spec.max_seq_len = 150
from spirl.utils.general_utils import AttrDict from spirl.data.kitchen.src.kitchen_data_loader import D4RLSequenceSplitDataset data_spec = AttrDict( dataset_class=D4RLSequenceSplitDataset, n_actions=9, state_dim=60, env_name="kitchen-mixed-v0", res=128, crop_rand_subseq=True, ) data_spec.max_seq_len = 280
from spirl.utils.general_utils import AttrDict from spirl.components.data_loader import GlobalSplitVideoDataset data_spec = AttrDict( dataset_class=GlobalSplitVideoDataset, n_actions=8, state_dim=97, split=AttrDict(train=0.9, val=0.1, test=0.0), env_name="Widow250OfficeFixed-v0", res=64, crop_rand_subseq=True, ) data_spec.max_seq_len = 350