Merge pull request #224 from ASUS-AICS/predict

Set data_loader_idx to 0 in predict_step()

Author: Eleven1Liu

libmultilabel/nn/model.py

@@ -152,7 +152,7 @@ def _shared_eval_epoch_end(self, step_outputs, split):
         self.eval_metric.reset()
         return metric_dict
 
-    def predict_step(self, batch, batch_idx, dataloader_idx):
+    def predict_step(self, batch, batch_idx, dataloader_idx=0):
         """`predict_step` is triggered when calling `trainer.predict()`.
         This function is used to get the top-k labels and their prediction scores.