Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Tensorboard out of try, iou_t to 0.10

This commit is contained in:
Glenn Jocher 2020-04-20 09:57:15 -07:00
parent accce6b565
commit be3f322375
1 changed files with 5 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
train.py
View File

@ -3,6 +3,7 @@ import argparse
import torch.distributed as dist import torch.distributed as dist
import torch.optim as optim import torch.optim as optim
import torch.optim.lr_scheduler as lr_scheduler import torch.optim.lr_scheduler as lr_scheduler
from torch.utils.tensorboard import SummaryWriter
import test # import test.py to get mAP after each epoch import test # import test.py to get mAP after each epoch
from models import * from models import *
@ -13,7 +14,7 @@ mixed_precision = True
try: # Mixed precision training https://github.com/NVIDIA/apex try: # Mixed precision training https://github.com/NVIDIA/apex
from apex import amp from apex import amp
except: except:
print('Apex recommended for mixed precision and faster training: https://github.com/NVIDIA/apex') print('Apex recommended for faster mixed precision training: https://github.com/NVIDIA/apex')
mixed_precision = False # not installed mixed_precision = False # not installed
wdir = 'weights' + os.sep # weights dir wdir = 'weights' + os.sep # weights dir
@ -28,7 +29,7 @@ hyp = {'giou': 3.54, # giou loss gain
'cls_pw': 1.0, # cls BCELoss positive_weight 'cls_pw': 1.0, # cls BCELoss positive_weight
'obj': 64.3, # obj loss gain (*=img_size/320 if img_size != 320) 'obj': 64.3, # obj loss gain (*=img_size/320 if img_size != 320)
'obj_pw': 1.0, # obj BCELoss positive_weight 'obj_pw': 1.0, # obj BCELoss positive_weight
'iou_t': 0.225, # iou training threshold 'iou_t': 0.1, # iou training threshold
'lr0': 0.01, # initial learning rate (SGD=5E-3, Adam=5E-4) 'lr0': 0.01, # initial learning rate (SGD=5E-3, Adam=5E-4)
'lrf': 0.0005, # final learning rate (with cos scheduler) 'lrf': 0.0005, # final learning rate (with cos scheduler)
'momentum': 0.937, # SGD momentum 'momentum': 0.937, # SGD momentum
@ -418,15 +419,8 @@ if __name__ == '__main__':
tb_writer = None tb_writer = None
if not opt.evolve: # Train normally if not opt.evolve: # Train normally
try: print('Start Tensorboard with "tensorboard --logdir=runs", view at http://localhost:6006/')
# Start Tensorboard with "tensorboard --logdir=runs", view at http://localhost:6006/ tb_writer = SummaryWriter()
from torch.utils.tensorboard import SummaryWriter
tb_writer = SummaryWriter()
print("Run 'tensorboard --logdir=runs' to view tensorboard at http://localhost:6006/")
except:
pass
train() # train normally train() # train normally
else: # Evolve hyperparameters (optional) else: # Evolve hyperparameters (optional)