Fixed display of running loss

Yolo_train.py

@@ -53,8 +53,8 @@ def train_one_epoch(yolo_network, train_loader):
         # print statistics
         running_loss.append(loss.item())
 
-        if i % 500 == 499: # print every 500 mini-batches
-            print(f'[{epoch + 1}, {i + 1:5d}] loss: {sum(running_loss) / 2000:.3f}')
+        if i % 200 == 199: # print every 200 mini-batches
+            print(f'[{epoch + 1}, {i + 1:5d}] loss: {sum(running_loss) / 200:.3f}')
             running_loss = []