MCPcopy
hub / github.com/deepspeedai/DeepSpeed / __init__

Method __init__

tests/unit/simple_model.py:71–72  ·  view source on GitHub ↗
(self, hidden_dim, empty_grad=False)

Source from the content-addressed store, hash-verified

69class Curriculum_SimpleModel(SimpleModel):
70
71 def __init__(self, hidden_dim, empty_grad=False):
72 super(Curriculum_SimpleModel, self).__init__(hidden_dim, empty_grad)
73
74 def forward(self, x, y, **kwargs):
75 seqlen = kwargs.get('curriculum_seqlen', None)

Callers

nothing calls this directly

Calls 1

__init__Method · 0.45

Tested by

no test coverage detected