MCPcopy Create free account
hub / github.com/Sin3DM/Sin3DM / ModelMeanType

Class ModelMeanType

src/diffusion/gaussian_diffusion.py:66–73  ·  view source on GitHub ↗

Which type of output the model predicts.

Source from the content-addressed store, hash-verified

64
65
66class ModelMeanType(enum.Enum):
67 """
68 Which type of output the model predicts.
69 """
70
71 PREVIOUS_X = enum.auto() # the model predicts x_{t-1}
72 START_X = enum.auto() # the model predicts x_0
73 EPSILON = enum.auto() # the model predicts epsilon
74
75
76class ModelVarType(enum.Enum):

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected