other reverts

supervisely-ecosystem · May 22, 2023 · c260377 · c260377
1 parent 39149dc
commit c260377
Show file tree

Hide file tree

Showing 4 changed files with 0 additions and 4 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -5,7 +5,6 @@ opencv-python-headless==4.5.5.62
 opencv-python==4.5.5.62
 
 --extra-index-url https://download.pytorch.org/whl/cu113
-# torch==2.0.0
 torch==1.11.0
 torchvision==0.12.0
 

diff --git a/train.py b/train.py
@@ -68,7 +68,6 @@
 
 def train(hyp, opt, device, tb_writer=None):
 
-    # device = torch.device("cpu")
     train_batches_uploaded = False
 
     logger.info("hyperparameters", extra=hyp)

diff --git a/utils/datasets.py b/utils/datasets.py
@@ -455,7 +455,6 @@ def cache_labels(self, path=Path('./labels.cache'), prefix=''):
         # Cache dataset labels, check images and read shapes
         x = {}  # dict
         nm, nf, ne, nc = 0, 0, 0, 0  # number missing, found, empty, duplicate
-        # possible bug
         pbar = tqdm(zip(self.img_files, self.label_files), desc='Scanning images', total=len(self.img_files))
         for i, (im_file, lb_file) in enumerate(pbar):
             try:

diff --git a/utils/loss.py b/utils/loss.py
@@ -161,7 +161,6 @@ def __call__(self, p, targets):  # predictions, targets, model
         return loss * bs, torch.cat((lbox, lobj, lcls, loss)).detach()
 
     def build_targets(self, p, targets):
-        # targets = targets.float()
         # Build targets for compute_loss(), input targets(image,class,x,y,w,h)
         na, nt = self.na, targets.shape[0]  # number of anchors, targets
         tcls, tbox, indices, anch = [], [], [], []