henry000 commited on
Commit
66f3f62
·
1 Parent(s): 1528a4f

💬 [Add] TODO reduce memory usage when cal loss

Browse files
Files changed (1) hide show
  1. yolo/tools/data_loader.py +1 -0
yolo/tools/data_loader.py CHANGED
@@ -189,6 +189,7 @@ class YoloDataLoader(DataLoader):
189
  batch_size = len(batch)
190
  target_sizes = [item[1].size(0) for item in batch]
191
  # TODO: Improve readability of these proccess
 
192
  batch_targets = torch.zeros(batch_size, min(max(target_sizes), 100), 5)
193
  batch_targets[:, :, 0] = -1
194
  for idx, target_size in enumerate(target_sizes):
 
189
  batch_size = len(batch)
190
  target_sizes = [item[1].size(0) for item in batch]
191
  # TODO: Improve readability of these proccess
192
+ # TODO: remove maxBbox or reduce loss function memory usage
193
  batch_targets = torch.zeros(batch_size, min(max(target_sizes), 100), 5)
194
  batch_targets[:, :, 0] = -1
195
  for idx, target_size in enumerate(target_sizes):