compute_num_train_frames

lightning_pose.data.utils.compute_num_train_frames(len_train_dataset: int, train_frames: int | float | None = None) → int

Quickly compute number of training frames for a given dataset.

Parameters:

• len_train_dataset – total number of frames in training dataset

• train_frames – <=1 - fraction of total train frames used for training >1 - number of total train frames used for training

Returns:

total number of train frames