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