pytouch/models/slip_detect.py [34:42]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
            self._model = model(**kwargs)
            self._model.fc = nn.Linear(
                self._model.fc.in_features, self.defaults.CLASSES
            )
        else:
            raise NotImplementedError()

    def _load_state_dict(self):
        self._model.load_state_dict(self.state_dict)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



pytouch/models/touch_detect.py [42:50]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
            self._model = model(**kwargs)
            self._model.fc = nn.Linear(
                self._model.fc.in_features, self.defaults.CLASSES
            )
        else:
            raise NotImplementedError()

    def _load_state_dict(self):
        self._model.load_state_dict(self.state_dict)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -