recovery

a3magic3pocket · Mar 9, 2023 · 022a849 · 022a849
1 parent c8de73b
commit 022a849
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 6 deletions.
diff --git a/dataset.py b/dataset.py
@@ -50,7 +50,6 @@ def __getitem__(self, index):
             buf.seek(0)
             try:
                 img = Image.open(buf).convert('L')
-                img.save(f"/content/drive/MyDrive/weights/crnn/qwerasdf{index}.jpg")
             except IOError:
                 print('Corrupted image for %d' % index)
                 return self[index + 1]
@@ -60,7 +59,6 @@ def __getitem__(self, index):
 
             label_key = 'label-%09d' % index
             label = txn.get(label_key.encode()).decode('utf-8')
-            print(label, 'label')
 
             if self.target_transform is not None:
                 label = self.target_transform(label)

diff --git a/my_train.py b/my_train.py
@@ -204,10 +204,6 @@ def trainBatch(net, criterion, optimizer):
     global length
     data = next(train_iter)
     cpu_images, cpu_texts = data
-    print('cpu_images', cpu_images)
-    import sys
-    sys.exit(1)
-
     batch_size = cpu_images.size(0)
     copied_image = utils.loadData(image, cpu_images)