BENDRClassifier Error #73
Description
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!