mirrors
/
yolov5
mirror of https://github.com/ultralytics/yolov5.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
yolov5/models
History
Yonghye Kwon a1c3572bc9
Explicit `model.eval()` call `if opt.train=False` (#3475) 
* call model.eval() when opt.train is False

call model.eval() when opt.train is False

* single-line if statement

* cleanup

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 		2021-06-06 18:39:36 +02:00
..
hub Add TransformerLayer, TransformerBlock, C3TR modules (#2333) 2021-04-01 17:26:53 +02:00
__init__.py add "__init__.py" file in <models> folder 2020-06-15 14:37:07 +08:00
common.py Parameterize max_det + inference default at 1000 (#3215) 2021-05-17 22:47:44 +02:00
experimental.py Enable direct `--weights URL` definition (#3373) 2021-05-28 15:18:44 +02:00
export.py Explicit `model.eval()` call `if opt.train=False` (#3475) 2021-06-06 18:39:36 +02:00
yolo.py Update FLOPs description (#3422) 2021-06-03 12:31:51 +02:00
yolov5l.yaml YOLOv5 v4.0 Release (#1837) 2021-01-04 19:54:09 -08:00
yolov5m.yaml YOLOv5 v4.0 Release (#1837) 2021-01-04 19:54:09 -08:00
yolov5s.yaml YOLOv5 v4.0 Release (#1837) 2021-01-04 19:54:09 -08:00
yolov5x.yaml YOLOv5 v4.0 Release (#1837) 2021-01-04 19:54:09 -08:00