student_specialization/model_gen.py [135:146]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    def scale(self, r):
        def _scale(w):
            w.weight.data *= r
            w.bias.data *= r

        for w in self.ws_linear:
            _scale(w)

        _scale(self.final_w)

    def forward(self, x):
        hs = []
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



student_specialization/model_gen.py [241:252]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    def scale(self, r):
        def _scale(w):
            w.weight.data *= r
            w.bias.data *= r

        for w in self.ws_linear:
            _scale(w)

        _scale(self.final_w)

    def forward(self, x):
        hs = []
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -