Index

_

__init__() (ActionConditionalSpatioTemporalLSTMCell method) (Activation method) (Autoencoder method) (BAIRPushingDataset method) (BaseConvRNN method) (CaltechPedestrianDataset method) (ConvLSTM method), [1] (ConvLSTMCell method) (CopyLastFrame method) (DCGANConv method) (DCGANConvTranspose method) (DCGANDecoder method) (DCGANEncoder method) (Decoder method) (DecoderSplit method) (DoubleConv2d method) (DoubleConv3d method) (EF_ConvLSTM method) (EF_TrajGRU method) (Encoder method), [1] (Encoder_Forecaster method) (EncoderSplit method) (Forecaster method) (FrechetVideoDistance method) (Human36MDataset method) (K2M method) (KITTIRawDataset method) (KTHActionsDataset method) (L1 method) (LPIPS method) (LSTM method) (MovingMNISTDataset method) (MovingMNISTOnTheFly method) (MSE method) (PhyCell method) (PhyCell_Cell method) (PhyDNet method) (Physics101Dataset method) (PredictionLossProvider method) (PredictionMetricProvider method) (PredRNN_V2 method) (PSNR method) (ScaleToModel method) (ScaleToTest method) (SingleStepConvLSTM method) (SmoothL1 method) (SpatioTemporalLSTMCell method) (SSIM method) (STPhy method) (SynpickMovingDataset method) (TrajGRU method) (UNet3D method) (VPDataset method) (VPDatasetWrapper method) (VPMeasure method) (VPModel method) (VPSuite method)

A

action_conditional (VPModel attribute) ACTION_SIZE (BAIRPushingDataset attribute) (CaltechPedestrianDataset attribute) (Human36MDataset attribute) (KITTIRawDataset attribute) (KTHActionsDataset attribute) (MovingMNISTDataset attribute) (MovingMNISTOnTheFly attribute) (Physics101Dataset attribute) (SynpickMovingDataset attribute) (VPDataset attribute) action_size (VPDatasetWrapper property) (VPModel attribute) ActionConditionalSpatioTemporalLSTMCell (class in vp_suite.model_blocks.predrnn)

actions (VPData attribute) Activation (class in vp_suite.model_blocks.traj_gru) activation (EF_TrajGRU attribute) add_border_around_vid() (in module vp_suite.utils.visualization) add_borders() (in module vp_suite.utils.visualization) alarm_handler() (in module vp_suite.utils.utils) ALL_SCENARIOS (Human36MDataset attribute) ALLOWED_SPLITS (VPDatasetWrapper attribute) arr_from_img() (in module vp_suite.datasets.mmnist) Autoencoder (class in vp_suite.model_blocks.enc) AVAILABLE_CAMERAS (KITTIRawDataset attribute) (Physics101Dataset attribute) available_metrics (PredictionMetricProvider attribute) AVAILABLE_SUBSEQ (Physics101Dataset attribute)

B

BAIRPushingDataset (class in vp_suite.datasets.bair) BaseConvRNN (class in vp_suite.model_blocks.traj_gru) batch_size (DefaultRunConfig attribute) BIGGER_IS_BETTER (PSNR attribute) (SSIM attribute) (VPMeasure attribute)

bottleneck_dim (LSTM attribute) build_models() (Autoencoder method)

C

calculate_2_wasserstein_dist() (in module vp_suite.measure.fvd.fvd) calculate_n_chunks() (FrechetVideoDistance method) CaltechPedestrianDataset (class in vp_suite.datasets.caltech_pedestrian) camera (KITTIRawDataset attribute) (Physics101Dataset attribute) CAN_HANDLE_ACTIONS (LSTM attribute) (PhyDNet attribute) (PredRNN_V2 attribute) (STPhy attribute) (UNet3D attribute) (VPModel attribute) check_model_and_data_compat() (in module vp_suite.utils.compatibility) check_optuna_config() (in module vp_suite.utils.utils) check_run_and_model_compat() (in module vp_suite.utils.compatibility) CLASSES (KTHActionsDataset attribute) clear_datasets() (VPSuite method) clear_models() (VPSuite method) CODE_REFERENCE (ActionConditionalSpatioTemporalLSTMCell attribute) (ConvLSTM attribute), [1] (EF_ConvLSTM attribute) (EF_TrajGRU attribute) (LSTM attribute) (PhyCell attribute) (PhyCell_Cell attribute) (PhyDNet attribute) (PredRNN_V2 attribute) (SpatioTemporalLSTMCell attribute) (TrajGRU attribute) (VPModel attribute) (VPModelBlock attribute)

COLORS (in module vp_suite.utils.visualization) config (VPDataset property) (VPDatasetWrapper property) (VPModel property) context_frames (DefaultRunConfig attribute) conv_actions_on_input (PredRNN_V2 attribute) conv_output_shape() (in module vp_suite.utils.models) ConvLSTM (class in vp_suite.model_blocks.conv_lstm_hzzone) (class in vp_suite.model_blocks.conv_lstm_ndrplz) convlstm_hidden_dims (PhyDNet attribute) convlstm_kernel_size (PhyDNet attribute) convlstm_n_layers (PhyDNet attribute) ConvLSTMCell (class in vp_suite.model_blocks.conv_lstm_ndrplz) convtransp_output_shape() (in module vp_suite.utils.models) CopyLastFrame (class in vp_suite.models.copy_last_frame) create_model() (VPSuite method)

D

data_dir (VPDataset attribute) (VPDatasetWrapper property) dataset (VPSubset attribute) DATASET_FRAME_SHAPE (BAIRPushingDataset attribute) (CaltechPedestrianDataset attribute) (Human36MDataset attribute) (KITTIRawDataset attribute) (KTHActionsDataset attribute) (MovingMNISTDataset attribute) (MovingMNISTOnTheFly attribute) (Physics101Dataset attribute) (SynpickMovingDataset attribute) (VPDataset attribute) datasets (VPSuite attribute) DCGANConv (class in vp_suite.model_blocks.conv) DCGANConvTranspose (class in vp_suite.model_blocks.conv) DCGANDecoder (class in vp_suite.model_blocks.enc) DCGANEncoder (class in vp_suite.model_blocks.enc) dec_c (EF_ConvLSTM attribute) (EF_TrajGRU attribute) dec_conv_k (EF_ConvLSTM attribute) (EF_TrajGRU attribute) dec_conv_names (EF_ConvLSTM attribute) (EF_TrajGRU attribute) dec_conv_p (EF_ConvLSTM attribute) (EF_TrajGRU attribute) dec_conv_s (EF_ConvLSTM attribute) (EF_TrajGRU attribute) dec_rnn_h2h_d (EF_TrajGRU attribute) dec_rnn_h2h_k (EF_TrajGRU attribute) dec_rnn_i2h_k (EF_TrajGRU attribute) dec_rnn_i2h_p (EF_TrajGRU attribute) dec_rnn_i2h_s (EF_TrajGRU attribute) dec_rnn_k (EF_ConvLSTM attribute) dec_rnn_L (EF_TrajGRU attribute) dec_rnn_p (EF_ConvLSTM attribute) dec_rnn_s (EF_ConvLSTM attribute) dec_rnn_z (EF_TrajGRU attribute) decode() (Autoencoder method) (LSTM method)

Decoder (class in vp_suite.model_blocks.enc) DecoderSplit (class in vp_suite.model_blocks.phydnet) decoupling_loss_scale (PredRNN_V2 attribute) (STPhy attribute) default_available() (VPDataset method) DEFAULT_DATA_DIR (BAIRPushingDataset attribute) (CaltechPedestrianDataset attribute) (Human36MDataset attribute) (KITTIRawDataset attribute) (KTHActionsDataset attribute) (MovingMNISTDataset attribute) (MovingMNISTOnTheFly attribute) (Physics101Dataset attribute) (SynpickMovingDataset attribute) (VPDataset attribute) DEFAULT_N_SEQS (MovingMNISTOnTheFly attribute) DEFAULT_RUN_CONFIG (in module vp_suite.defaults) DefaultRunConfig (class in vp_suite.defaults) device (PredictionLossProvider attribute) (PredictionMetricProvider attribute) (VPMeasure attribute) (VPSuite attribute) dim() (K2M method) DoubleConv2d (class in vp_suite.model_blocks.conv) DoubleConv3d (class in vp_suite.model_blocks.conv) download_and_extract_bair() (in module vp_suite.datasets.bair) download_and_prepare_dataset() (BAIRPushingDataset class method) (CaltechPedestrianDataset class method) (Human36MDataset class method) (KITTIRawDataset class method) (KTHActionsDataset class method) (MovingMNISTDataset method) (MovingMNISTOnTheFly method) (Physics101Dataset method) (SynpickMovingDataset method) (VPDataset class method) download_dataset() (VPSuite method) download_from_url() (in module vp_suite.utils.utils) download_synpick() (in module vp_suite.datasets.synpick)

E

EF_ConvLSTM (class in vp_suite.models.precipitation_nowcasting.ef_conv_lstm) EF_TrajGRU (class in vp_suite.models.precipitation_nowcasting.ef_traj_gru) enc_c (EF_ConvLSTM attribute) (EF_TrajGRU attribute) enc_conv_k (EF_ConvLSTM attribute) (EF_TrajGRU attribute) enc_conv_names (EF_ConvLSTM attribute) (EF_TrajGRU attribute) enc_conv_p (EF_ConvLSTM attribute) (EF_TrajGRU attribute) enc_conv_s (EF_ConvLSTM attribute) (EF_TrajGRU attribute) enc_rnn_h2h_d (EF_TrajGRU attribute) enc_rnn_h2h_k (EF_TrajGRU attribute) enc_rnn_i2h_k (EF_TrajGRU attribute)

enc_rnn_i2h_p (EF_TrajGRU attribute) enc_rnn_i2h_s (EF_TrajGRU attribute) enc_rnn_k (EF_ConvLSTM attribute) enc_rnn_L (EF_TrajGRU attribute) enc_rnn_p (EF_ConvLSTM attribute) enc_rnn_s (EF_ConvLSTM attribute) enc_rnn_z (EF_TrajGRU attribute) encode() (Autoencoder method) (LSTM method) Encoder (class in vp_suite.model_blocks.enc) (class in vp_suite.models.precipitation_nowcasting.ef_blocks) Encoder_Forecaster (class in vp_suite.models.precipitation_nowcasting.ef_blocks) encoder_fwd() (PhyDNet method) EncoderSplit (class in vp_suite.model_blocks.phydnet) epochs (DefaultRunConfig attribute) eval_iter() (VPModel method)

F

features (UNet3D attribute) filter_size (PredRNN_V2 attribute) final_conv_1_c (EF_ConvLSTM attribute) (EF_TrajGRU attribute) final_conv_1_k (EF_ConvLSTM attribute) (EF_TrajGRU attribute) final_conv_1_name (EF_ConvLSTM attribute) (EF_TrajGRU attribute) final_conv_1_p (EF_ConvLSTM attribute) (EF_TrajGRU attribute) final_conv_1_s (EF_ConvLSTM attribute) (EF_TrajGRU attribute) final_conv_2_k (EF_ConvLSTM attribute) (EF_TrajGRU attribute) final_conv_2_name (EF_ConvLSTM attribute) (EF_TrajGRU attribute) final_conv_2_p (EF_ConvLSTM attribute) (EF_TrajGRU attribute) final_conv_2_s (EF_ConvLSTM attribute) (EF_TrajGRU attribute) find_divisor_for_group_norm() (in module vp_suite.model_blocks.phydnet) first_frame_rng_seed (KTHActionsDataset attribute) Forecaster (class in vp_suite.models.precipitation_nowcasting.ef_blocks) forward() (ActionConditionalSpatioTemporalLSTMCell method) (ConvLSTM method), [1] (ConvLSTMCell method) (DCGANConv method) (DCGANConvTranspose method) (DCGANDecoder method) (DCGANEncoder method) (Decoder method) (DecoderSplit method) (DoubleConv2d method) (DoubleConv3d method) (Encoder method), [1] (Encoder_Forecaster method) (EncoderSplit method) (Forecaster method) (FrechetVideoDistance method) (K2M method) (LPIPS method) (LSTM method) (PhyCell method) (PhyCell_Cell method) (PhyDNet method) (PredRNN_V2 method) (PSNR method) (ScaleToModel method) (ScaleToTest method) (SingleStepConvLSTM method) (SpatioTemporalLSTMCell method) (SSIM method) (STPhy method) (TrajGRU method) (UNet3D method) (VPMeasure method) (VPModel method)

forward_by_stage() (Encoder method) (Forecaster method) FPS (CaltechPedestrianDataset attribute) (Human36MDataset attribute) (KITTIRawDataset attribute) frames (VPData attribute) FrechetVideoDistance (class in vp_suite.measure.fvd.fvd)

G

generate_moving_mnist() (in module vp_suite.datasets.mmnist) get_color_array() (in module vp_suite.utils.visualization) get_distance() (FrechetVideoDistance method) get_frame_count() (in module vp_suite.utils.utils) get_from_idx() (KTHActionsDataset method)

get_losses() (PredictionLossProvider method) get_metrics() (PredictionMetricProvider method) get_public_attrs() (in module vp_suite.utils.utils) get_test() (VPDataset class method) get_train_val() (VPDataset class method) get_vis_from_model() (in module vp_suite.utils.visualization)

H

Human36MDataset (class in vp_suite.datasets.human36m)

hyperopt() (VPSuite method)

I

img_from_arr() (in module vp_suite.datasets.mmnist) img_shape (VPDataset attribute) (VPDatasetWrapper property) (VPModel attribute) indices (VPSubset attribute) inflated_action_dim (PredRNN_V2 attribute) (STPhy attribute) init_hidden() (ConvLSTMCell method) (PhyCell method) (SingleStepConvLSTM method) invM (K2M property) IS_DOWNLOADABLE (BAIRPushingDataset attribute) (CaltechPedestrianDataset attribute) (Human36MDataset attribute) (KITTIRawDataset attribute) (KTHActionsDataset attribute) (MovingMNISTDataset attribute) (MovingMNISTOnTheFly attribute) (Physics101Dataset attribute) (SynpickMovingDataset attribute) (VPDataset attribute)

is_model_block() (in module vp_suite.model_blocks) is_test_set() (VPDatasetWrapper method) is_training_set() (VPDatasetWrapper method)

K

K2M (class in vp_suite.model_blocks.phydnet)

KITTIRawDataset (class in vp_suite.datasets.kitti_raw) KTHActionsDataset (class in vp_suite.datasets.kth)

L

L1 (class in vp_suite.measure.image_wise) layer_norm (PredRNN_V2 attribute) list_available_datasets() (VPSuite method) list_available_models() (VPSuite method) load_dataset() (in module vp_suite.datasets.mmnist) (VPSuite method) load_model() (VPSuite method)

LOSS_CLASSES (in module vp_suite.measure) losses (PredictionLossProvider attribute) losses_and_scales (DefaultRunConfig attribute) LPIPS (class in vp_suite.measure.image_wise) lr (DefaultRunConfig attribute) LSTM (class in vp_suite.models.lstm) lstm_hidden_dim (LSTM attribute) lstm_num_layers (LSTM attribute)

M

M (K2M property) MATCHES_REFERENCE (ActionConditionalSpatioTemporalLSTMCell attribute) (ConvLSTM attribute), [1] (EF_ConvLSTM attribute) (EF_TrajGRU attribute) (LSTM attribute) (PhyCell attribute) (PhyCell_Cell attribute) (PhyDNet attribute) (PredRNN_V2 attribute) (SpatioTemporalLSTMCell attribute) (TrajGRU attribute) (VPModel attribute) (VPModelBlock attribute) max_acc (MovingMNISTOnTheFly attribute) max_speed (MovingMNISTOnTheFly attribute) max_training_hours (DefaultRunConfig attribute) METRIC_CLASSES (in module vp_suite.measure) metrics (DefaultRunConfig attribute) (PredictionMetricProvider attribute) min_acc (MovingMNISTOnTheFly attribute) MIN_CONTEXT_FRAMES (VPModel attribute) MIN_SEQ_LEN (BAIRPushingDataset attribute) (CaltechPedestrianDataset attribute) (Human36MDataset attribute) (KITTIRawDataset attribute) (KTHActionsDataset attribute) (MovingMNISTOnTheFly attribute) (Physics101Dataset attribute) (SynpickMovingDataset attribute) (VPDataset attribute) min_speed (MovingMNISTOnTheFly attribute) MODEL_CLASSES (in module vp_suite.models) model_dir (VPModel attribute) models (VPSuite attribute) module vp_suite vp_suite.base vp_suite.base.base_dataset vp_suite.base.base_measure vp_suite.base.base_model vp_suite.base.base_model_block vp_suite.datasets vp_suite.datasets.bair vp_suite.datasets.caltech_pedestrian vp_suite.datasets.human36m vp_suite.datasets.kitti_raw vp_suite.datasets.kth vp_suite.datasets.mmnist vp_suite.datasets.mmnist_on_the_fly vp_suite.datasets.physics101 vp_suite.datasets.synpick vp_suite.defaults vp_suite.measure vp_suite.measure.fvd vp_suite.measure.fvd.fvd vp_suite.measure.image_wise vp_suite.measure.loss_provider vp_suite.measure.metric_provider vp_suite.model_blocks vp_suite.model_blocks.conv vp_suite.model_blocks.conv_lstm_hzzone vp_suite.model_blocks.conv_lstm_ndrplz vp_suite.model_blocks.enc vp_suite.model_blocks.phydnet vp_suite.model_blocks.predrnn vp_suite.model_blocks.traj_gru vp_suite.models vp_suite.models.copy_last_frame vp_suite.models.lstm vp_suite.models.phydnet vp_suite.models.precipitation_nowcasting vp_suite.models.precipitation_nowcasting.ef_blocks vp_suite.models.precipitation_nowcasting.ef_conv_lstm vp_suite.models.precipitation_nowcasting.ef_traj_gru vp_suite.models.predrnn_v2 vp_suite.models.st_phy vp_suite.models.unet3d vp_suite.utils vp_suite.utils.compatibility vp_suite.utils.dataset_wrapper vp_suite.utils.models vp_suite.utils.utils vp_suite.utils.visualization vp_suite.vpsuite

moment_loss_scale (PhyDNet attribute) (STPhy attribute) most() (in module vp_suite.utils.utils) MovingMNISTDataset (class in vp_suite.datasets.mmnist) MovingMNISTOnTheFly (class in vp_suite.datasets.mmnist_on_the_fly) MSE (class in vp_suite.measure.image_wise)

N

n_seqs (MovingMNISTOnTheFly attribute) n_vis (DefaultRunConfig attribute) NAME (ActionConditionalSpatioTemporalLSTMCell attribute) (Autoencoder attribute) (BAIRPushingDataset attribute) (CaltechPedestrianDataset attribute) (ConvLSTM attribute), [1] (CopyLastFrame attribute) (DCGANConv attribute) (DCGANConvTranspose attribute) (DCGANDecoder attribute) (DCGANEncoder attribute) (Decoder attribute) (DoubleConv2d attribute) (DoubleConv3d attribute) (EF_ConvLSTM attribute) (EF_TrajGRU attribute) (Encoder attribute) (Encoder_Forecaster attribute) (FrechetVideoDistance attribute) (Human36MDataset attribute) (KITTIRawDataset attribute) (KTHActionsDataset attribute) (L1 attribute) (LPIPS attribute) (LSTM attribute) (MovingMNISTDataset attribute) (MovingMNISTOnTheFly attribute) (MSE attribute) (PhyCell attribute) (PhyCell_Cell attribute) (PhyDNet attribute) (Physics101Dataset attribute) (PredRNN_V2 attribute) (PSNR attribute) (SmoothL1 attribute) (SpatioTemporalLSTMCell attribute) (SSIM attribute) (STPhy attribute) (SynpickMovingDataset attribute) (TrajGRU attribute) (UNet3D attribute) (VPDataset attribute) (VPDatasetWrapper property) (VPMeasure attribute) (VPModel attribute) (VPModelBlock attribute)

NEEDS_COMPLETE_INPUT (PredRNN_V2 attribute) (VPModel attribute) no_train (DefaultRunConfig attribute) no_val (DefaultRunConfig attribute) no_vis (DefaultRunConfig attribute) no_wandb (DefaultRunConfig attribute) NON_CONFIG_VARS (VPDataset attribute) (VPModel attribute) num_channels (MovingMNISTOnTheFly attribute) num_digits (MovingMNISTOnTheFly attribute) num_hidden (PredRNN_V2 attribute) num_layers (EF_ConvLSTM attribute) (EF_TrajGRU attribute) (PredRNN_V2 attribute) (STPhy attribute)

O

ON_THE_FLY (MovingMNISTOnTheFly attribute) (VPDataset attribute) OPT_VALUE (PSNR attribute) (SSIM attribute) (VPMeasure attribute)

origin (VPData attribute) out_dir (DefaultRunConfig attribute)

P

PAPER_REFERENCE (ActionConditionalSpatioTemporalLSTMCell attribute) (ConvLSTM attribute) (DCGANConv attribute) (DCGANConvTranspose attribute) (DCGANDecoder attribute) (DCGANEncoder attribute) (DoubleConv2d attribute) (EF_ConvLSTM attribute) (EF_TrajGRU attribute) (LSTM attribute) (PhyCell attribute) (PhyCell_Cell attribute) (PhyDNet attribute) (PredRNN_V2 attribute) (SpatioTemporalLSTMCell attribute) (TrajGRU attribute) (VPModel attribute) (VPModelBlock attribute) patch_size (PredRNN_V2 attribute) PhyCell (class in vp_suite.model_blocks.phydnet) PhyCell_Cell (class in vp_suite.model_blocks.phydnet) phycell_channels (PhyDNet attribute) (STPhy attribute)

phycell_kernel_size (PhyDNet attribute) (STPhy attribute) phycell_n_layers (PhyDNet attribute) PhyDNet (class in vp_suite.models.phydnet) Physics101Dataset (class in vp_suite.datasets.physics101) postprocess() (VPDataset method) pred_1() (CopyLastFrame method) (Encoder_Forecaster method) (LSTM method) (PhyDNet method) (PredRNN_V2 method) (STPhy method) (UNet3D method) (VPModel method) pred_frames (DefaultRunConfig attribute) PredictionLossProvider (class in vp_suite.measure.loss_provider) PredictionMetricProvider (class in vp_suite.measure.metric_provider) PredRNN_V2 (class in vp_suite.models.predrnn_v2) preprocess() (VPDataset method) PSNR (class in vp_suite.measure.image_wise) PytestExpectedException

R

r_exp_alpha (PredRNN_V2 attribute) r_sampling_step_1 (PredRNN_V2 attribute) r_sampling_step_2 (PredRNN_V2 attribute) read_video() (in module vp_suite.utils.utils) REFERENCE (BAIRPushingDataset attribute) (CaltechPedestrianDataset attribute) (FrechetVideoDistance attribute) (Human36MDataset attribute) (KITTIRawDataset attribute) (KTHActionsDataset attribute) (LPIPS attribute) (MovingMNISTDataset attribute) (Physics101Dataset attribute) (SSIM attribute) (SynpickMovingDataset attribute) (VPDataset attribute) (VPMeasure attribute)

REQUIRED_ARGS (CopyLastFrame attribute) (UNet3D attribute) (VPModel attribute) reset_rng() (MovingMNISTOnTheFly method) (VPDataset method) (VPDatasetWrapper method) (VPSuite method) reshape_clamp() (VPMeasure method) residual_on_action_conv (PredRNN_V2 attribute) reverse_input (PredRNN_V2 attribute) reverse_scheduled_sampling (PredRNN_V2 attribute) rng_seed (MovingMNISTOnTheFly attribute) run_shell_command() (in module vp_suite.utils.utils)

S

sampling_changing_rate (PredRNN_V2 attribute) sampling_eta (PredRNN_V2 attribute) sampling_stop_iter (PredRNN_V2 attribute) save_arr_hist() (in module vp_suite.utils.visualization) save_frame_compare_img() (in module vp_suite.utils.visualization) save_generated_mmnist() (in module vp_suite.datasets.mmnist) save_vid_vis() (in module vp_suite.utils.visualization) ScaleToModel (class in vp_suite.utils.models) ScaleToTest (class in vp_suite.utils.models) scenarios (Human36MDataset attribute) scheduled_sampling (PredRNN_V2 attribute) seed (DefaultRunConfig attribute) seq_step (DefaultRunConfig attribute) (VPDataset attribute) set_from_kwarg() (in module vp_suite.utils.utils) set_hidden() (SingleStepConvLSTM method) set_seq_len() (VPDataset method) (VPDatasetWrapper method)

SETTINGS (in module vp_suite.defaults) SHORT_CLASSES (KTHActionsDataset attribute) SingleStepConvLSTM (class in vp_suite.model_blocks.phydnet) size() (K2M method) SKIP_FIRST_N (Human36MDataset attribute) (SynpickMovingDataset attribute) SmoothL1 (class in vp_suite.measure.image_wise) SpatioTemporalLSTMCell (class in vp_suite.model_blocks.predrnn) split (VPDataset attribute) split_bair_traj_files() (in module vp_suite.datasets.bair) SPLIT_SEED_OFFSETS (MovingMNISTOnTheFly attribute) SSIM (class in vp_suite.measure.image_wise) st_cell_channels (STPhy attribute) state_dicts_equal() (in module vp_suite.utils.models) STPhy (class in vp_suite.models.st_phy) stride (PredRNN_V2 attribute) subseq (Physics101Dataset attribute) SynpickMovingDataset (class in vp_suite.datasets.synpick)

T

teacher_forcing_decay (PhyDNet attribute) (STPhy attribute) temporal_dim (UNet3D attribute) tensor_value_range (VPModel attribute) tensordot() (in module vp_suite.model_blocks.phydnet) test() (VPSuite method) test_data (VPDatasetWrapper property) TEST_SETS (CaltechPedestrianDataset attribute) test_sets (VPSuite property) timed_input() (in module vp_suite.utils.utils) TimeOutException timestamp() (in module vp_suite.utils.utils) to_display() (PSNR class method) (SSIM class method) (VPMeasure class method) TqdmUpTo (class in vp_suite.utils.utils) train() (VPSuite method) train_data (VPDatasetWrapper property) train_iter() (PhyDNet method) (PredRNN_V2 method) (STPhy method) (VPModel method) train_to_val_ratio (BAIRPushingDataset attribute) (CaltechPedestrianDataset attribute) (Human36MDataset attribute) (KITTIRawDataset attribute) (MovingMNISTDataset attribute) (SynpickMovingDataset attribute) (VPDataset attribute) train_val_seed (VPDataset attribute) TRAIN_VAL_SETS (CaltechPedestrianDataset attribute) TRAINABLE (CopyLastFrame attribute) (VPModel attribute) training (ActionConditionalSpatioTemporalLSTMCell attribute) (Autoencoder attribute) (BaseConvRNN attribute) (ConvLSTM attribute), [1] (ConvLSTMCell attribute) (CopyLastFrame attribute) (DCGANConv attribute) (DCGANConvTranspose attribute) (DCGANDecoder attribute) (DCGANEncoder attribute) (Decoder attribute) (DecoderSplit attribute) (DoubleConv2d attribute) (DoubleConv3d attribute) (EF_ConvLSTM attribute) (Encoder attribute), [1] (Encoder_Forecaster attribute) (EncoderSplit attribute) (Forecaster attribute) (FrechetVideoDistance attribute) (K2M attribute) (L1 attribute) (LPIPS attribute) (MSE attribute) (PhyCell attribute) (PhyCell_Cell attribute) (PSNR attribute) (ScaleToModel attribute) (ScaleToTest attribute) (SingleStepConvLSTM attribute) (SmoothL1 attribute) (SpatioTemporalLSTMCell attribute) (SSIM attribute) (STPhy attribute) (TrajGRU attribute) (UNet3D attribute)

training_iteration (PredRNN_V2 attribute) training_sets (VPSuite property) trainval_test_seed (KITTIRawDataset attribute) (Physics101Dataset attribute) trainval_to_test_ratio (KITTIRawDataset attribute) (Physics101Dataset attribute) TrajGRU (class in vp_suite.model_blocks.traj_gru) transform (VPDataset attribute)

U

UNet3D (class in vp_suite.models.unet3d) unpack_data() (VPModel method)

update_to() (TqdmUpTo method) use_actions (DefaultRunConfig attribute)

V

val_data (VPDatasetWrapper property) val_rec_criterion (DefaultRunConfig attribute) VALID_SPLITS (CaltechPedestrianDataset attribute) (Human36MDataset attribute) (KITTIRawDataset attribute) (MovingMNISTOnTheFly attribute) (SynpickMovingDataset attribute) (VPDataset attribute) value_range_max (VPDataset attribute) value_range_min (VPDataset attribute) vis_compare (DefaultRunConfig attribute) vis_context_frame_idx (DefaultRunConfig attribute) vis_every (DefaultRunConfig attribute) vis_mode (DefaultRunConfig attribute) visualize_sequences() (in module vp_suite.utils.visualization) visualize_vid() (in module vp_suite.utils.visualization) vp_suite module vp_suite.base module vp_suite.base.base_dataset module vp_suite.base.base_measure module vp_suite.base.base_model module vp_suite.base.base_model_block module vp_suite.datasets module vp_suite.datasets.bair module vp_suite.datasets.caltech_pedestrian module vp_suite.datasets.human36m module vp_suite.datasets.kitti_raw module vp_suite.datasets.kth module vp_suite.datasets.mmnist module vp_suite.datasets.mmnist_on_the_fly module vp_suite.datasets.physics101 module vp_suite.datasets.synpick module vp_suite.defaults module vp_suite.measure module vp_suite.measure.fvd module vp_suite.measure.fvd.fvd module vp_suite.measure.image_wise module vp_suite.measure.loss_provider module vp_suite.measure.metric_provider module

vp_suite.model_blocks module vp_suite.model_blocks.conv module vp_suite.model_blocks.conv_lstm_hzzone module vp_suite.model_blocks.conv_lstm_ndrplz module vp_suite.model_blocks.enc module vp_suite.model_blocks.phydnet module vp_suite.model_blocks.predrnn module vp_suite.model_blocks.traj_gru module vp_suite.models module vp_suite.models.copy_last_frame module vp_suite.models.lstm module vp_suite.models.phydnet module vp_suite.models.precipitation_nowcasting module vp_suite.models.precipitation_nowcasting.ef_blocks module vp_suite.models.precipitation_nowcasting.ef_conv_lstm module vp_suite.models.precipitation_nowcasting.ef_traj_gru module vp_suite.models.predrnn_v2 module vp_suite.models.st_phy module vp_suite.models.unet3d module vp_suite.utils module vp_suite.utils.compatibility module vp_suite.utils.dataset_wrapper module vp_suite.utils.models module vp_suite.utils.utils module vp_suite.utils.visualization module vp_suite.vpsuite module VPData (class in vp_suite.base.base_dataset) VPDataset (class in vp_suite.base.base_dataset) VPDatasetWrapper (class in vp_suite.utils.dataset_wrapper) VPMeasure (class in vp_suite.base.base_measure) VPModel (class in vp_suite.base.base_model) VPModelBlock (class in vp_suite.base.base_model_block) VPSubset (class in vp_suite.base.base_dataset) VPSuite (class in vp_suite.vpsuite)