changed learning rate

Yolo_train.py

@@ -33,7 +33,7 @@ yoloNetwork = Yolo_v1_fcs(1280*7*7)
 
 #Loss stuff:
 criterion = YoloLoss()
-optimizer = optim.SGD(yoloNetwork.parameters(), lr=0.001, momentum=0.9)
+optimizer = optim.SGD(yoloNetwork.parameters(), lr=0.0001, momentum=0.9)
 
 for epoch in range(2):
   running_loss = 0.0