Commit Graph
2 Commits
Author SHA1 Message Date
Felix YouandGlenn Jocher 361705d9be Multi-layer capable --freeze argument (#6019)
* support specfiy multiple frozen layers

* fix bug

* Cleanup Freeze section

* Cleanup argument

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-12-17 15:42:26 +01:00
Felix YouandGlenn Jocher c45f9f678d Make select_device() robust to batch_size=-1 (#5940)
* Find out a bug. When set batch_size = -1 to use the autobatch.

reproduce:

* Fix type conflict

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-12-10 15:27:20 +01:00