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
