spandrel.architectures.RGT.RGTArch

class documentation

class RGTArch(Architecture[RGT]): (source)

View In Hierarchy

Undocumented

Method __init__ Undocumented
Method load Loads the given state dict into a model. The hyperparameters will automatically be deduced.

Inherited from Architecture:

Method detect Inspects the given state dict and returns True if it is a state dict of this architecture.
Property id The unique ID of the architecture.
Property name The name of the architecture.
Instance Variable _detect Undocumented
Instance Variable _id Undocumented
Instance Variable _name Undocumented
def __init__(self): (source)
overrides spandrel.Architecture.__init__

Undocumented

@override
def load(self, state_dict: StateDict) -> ImageModelDescriptor[RGT]: (source)
overrides spandrel.Architecture.load

Loads the given state dict into a model. The hyperparameters will automatically be deduced.

The state dict is assumed to be a state dict of this architecture, meaning that detect returned True for the state dict. If this is not the case, then the behavior of this function is unspecified (the model may be loaded incorrect or an error is thrown).