File size: 470 Bytes
8652b14 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 | from datasets.video import (
MinecraftVideoDataset
)
from algorithms.worldmem import WorldMemMinecraft
from .exp_base import BaseLightningExperiment
class VideoPredictionExperiment(BaseLightningExperiment):
"""
A video prediction experiment
"""
compatible_algorithms = dict(
df_video_worldmemminecraft=WorldMemMinecraft,
)
compatible_datasets = dict(
# video datasets
video_minecraft=MinecraftVideoDataset,
)
|