Functions — github.com/bloodwass/mixout

CodeHub MCPcopy Create free account

hub / github.com/bloodwass/mixout / functions

Functions14 in github.com/bloodwass/mixout

⨍ Functions 14 ◇ Types & classes 3

↓ 2 callersMethod_make_noise

(input:torch.Tensor)

mixout/mixout.py:23

↓ 2 callersFunctiondemo

Demos applying Mixout to a pretrained model from Huggingface transformers params: model_name (str): the name of a pretra

example_huggingface.py:17

↓ 1 callersMethodbackward

(ctx, grad_output:torch.Tensor)

mixout/mixout.py:75

↓ 1 callersFunctionmain

↓ 1 callersFunctionmixout

(input:torch.Tensor, target:Optional["OrderedDict[str, torch.Tensor]"]=None, p:float=0

mixout/functional/__init__.py:7

↓ 1 callersFunctionrecursive_setattr

Recursively sets attributes for objects with children. params: obj (any) : the object with children wh

example_huggingface.py:61

↓ 1 callersFunctionreplace_layer_for_mixout

Replaces a single layer with the correct layer for use with Mixout. If module is nn.Dropout, replaces it with a Dropout where p = 0.

example_huggingface.py:30

↓ 1 callersMethodreset_parameters

mixout/mixlinear.py:48

(self, in_features:int, out_features:int, bias:bool=True,

mixout/mixlinear.py:25

Methodextra_repr

mixout/mixlinear.py:59

(self, input:torch.Tensor)

mixout/mixlinear.py:55

(cls, ctx, input:torch.Tensor, target:Optional["OrderedDict

mixout/mixout.py:27