💬 [Add] TODO reduce memory usage when cal loss
Browse files
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):
|