tempor.models.constants module¶

Common constants for the models package directory.

tempor.models.constants.DEVICE = device(type='cpu')¶

torch.device to use for computation, defaults to "cuda:0" if available, else "cpu".

tempor.models.constants.EPS : float = 1.1920928955078125e-07¶

Machine epsilon for torch.float32.

tempor.models.constants.ModelTaskType¶

Possible values specifying a model task type.

alias of Literal[classification, regression]

tempor.models.constants.ODEBackend¶

Possible values specifying an ODE backend.

alias of Literal[ode, cde, laplace]

tempor.models.constants.Nonlin¶

Possible values specifying a nonlinearity.

alias of Literal[none, elu, relu, leaky_relu, selu, tanh, sigmoid, softmax, gumbel_softmax]

tempor.models.constants.Samp¶

Possible values specifying a Sampler.

alias of Literal[BatchSampler, RandomSampler, Sampler, SequentialSampler, SubsetRandomSampler, WeightedRandomSampler]