DataFrame( index=test_dataset.samples_frame.id, columns=["proba", "label"] ) test_dataloader = torch.utils.data.DataLoader( test_dataset, shuffle=False, batch_size=self.hparams.get("batch_size", 4), num_workers=self.hparams.get("num_workers", 16)) for batch in tqdm(test_...