(self, hidden_dim, empty_grad=False)
| 69 | class 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) |