File size: 304 Bytes
604e535 | 1 2 3 4 5 6 7 8 | """FlowMo default config."""
from experiments.shared.src.models.image_world_models import ImageWorldModelConfig
def default_config():
"""Return the default FlowMo image-input config."""
return ImageWorldModelConfig(emb_dim=64, z_dim=112, c_dim=8, hidden_dim=128, history_len=8, context_len=32)
|