Method__call__Given batch frames with shape `[B, C, T, H, W]` extracted from a list of videos and a list of prompts (optional) correspondingly, return the
videox_fun/reward/reward_fn.py:24
Method__init__(
self,
csv_path, video_folder,
sample_size=256, sample_stride=4, sample_n
videox_fun/data/dataset_video.py:81
Method__init__(
self,
json_path, video_folder=None,
sample_size=256, sample_stride=4, sample_n_frame
videox_fun/data/dataset_video.py:158
Method__init__(
self,
dim,
*,
context_dim=None,
dim_head=64,
heads=12,
videox_fun/reward/MPS/trainer/models/cross_modeling.py:173
Method__init__(
self,
dim=512,
layer_num=4,
dim_head=64,
heads=8,
ff_mult=4
videox_fun/reward/MPS/trainer/models/cross_modeling.py:262
Method__init__(
self, rank: int, world_size: int, Controller,
GPU_memory_mode, scheduler_dict, model
videox_fun/api/api_multi_nodes.py:24