From 615e223aa55238c9ce0ba42aa44231a648454187 Mon Sep 17 00:00:00 2001
From: psj1997 <1176894699@qq.com>
Date: Tue, 19 Sep 2023 16:21:52 +0800
Subject: [PATCH] Fix issue when there is one sample in last batch

---
 SemiBin/semi_supervised_model.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/SemiBin/semi_supervised_model.py b/SemiBin/semi_supervised_model.py
index feca9b2..4543bb9 100644
--- a/SemiBin/semi_supervised_model.py
+++ b/SemiBin/semi_supervised_model.py
@@ -263,7 +263,8 @@ def train(out, contig_fastas, binned_lengths, logger, datas, data_splits, cannot
                 dataset=dataset_unlabeled,
                 batch_size=batchsize,
                 shuffle=True,
-                num_workers=0)
+                num_workers=0,
+                drop_last=True)
 
             for train_input1, train_input2, train_label in train_loader:
                 model.train()