Hello, I'm trying to follow the basic.ipynb walkthrough with the use of BENDR model provided from the bendr branch . The problem is on the process.fit() method and more specifically, on models.py line 116:

def forward(self, *x):
        features = self.features_forward(*x)
        if self.return_features:
            return self.classifier_forward(features), features
        else:
            return self.classifier_forward(features)

By adding the .t() pytorch method on the

return self.classifier_forward(features.t()), features

the error seems to be solved. I believe this may occur somewhere else also.

Moreover, the BENDR model seems to not apply the batch size configuration from .yml file. I set a batch size of 4 and an error occured. When setting to 11 (magic number) the model is training.

Thank you in advance for your time and the incorporation of BENDR on dn3 library!