tensorboard/focal loss reporting update
This commit is contained in:
		
							parent
							
								
									c7f93bae40
								
							
						
					
					
						commit
						05ae6e8499
					
				
							
								
								
									
										6
									
								
								train.py
								
								
								
								
							
							
						
						
									
										6
									
								
								train.py
								
								
								
								
							|  | @ -48,6 +48,10 @@ if f: | |||
|     for k, v in zip(hyp.keys(), np.loadtxt(f[0])): | ||||
|         hyp[k] = v | ||||
| 
 | ||||
| # Print focal loss if gamma > 0 | ||||
| if hyp['fl_gamma']: | ||||
|     print('Using FocalLoss(gamma=%g)' % hyp['fl_gamma']) | ||||
| 
 | ||||
| 
 | ||||
| def train(): | ||||
|     cfg = opt.cfg | ||||
|  | @ -412,7 +416,7 @@ if __name__ == '__main__': | |||
|             from torch.utils.tensorboard import SummaryWriter | ||||
| 
 | ||||
|             tb_writer = SummaryWriter() | ||||
|             print('Tensorboard started at http://localhost:6006/') | ||||
|             print("Tensorboard found. Run 'tensorboard --logdir=runs' to view at http://localhost:6006/") | ||||
|         except: | ||||
|             pass | ||||
| 
 | ||||
|  |  | |||
		Loading…
	
		Reference in New Issue