Skip to content

mps

Classes

fastvideo.platforms.mps.MpsPlatform

Bases: Platform

Functions

fastvideo.platforms.mps.MpsPlatform.seed_everything classmethod 
seed_everything(seed: int | None = None) -> None

Set the seed for MPS device.

Source code in fastvideo/platforms/mps.py 
65
66
67
68
69
70
71
72
73
74
@classmethod
def seed_everything(cls, seed: int | None = None) -> None:
    """Set the seed for MPS device."""
    if seed is not None:
        import random

        import numpy as np
        random.seed(seed)
        np.random.seed(seed)
        torch.manual_seed(seed)

Functions