Commit Graph

4 Commits

Author SHA1 Message Date
Mohammad Khoshbin
b8956dd5a5
fix training with frozen layers (#378) 2022-08-02 17:55:28 +03:00
Mohammad Khoshbin
cc42a206ef
fix cpu-only training CUDA error (#369) 2022-07-30 18:43:18 +03:00
Dhia_Oussayed
0d882e553e
Fixed a bug for Hyperparameters Evolution. (#344)
* Update hyp.scratch.custom.yaml

anchors parameter missing, evolution won't start without.

* Update train.py

Updated the hyperparameter evolution metadata variable to match the hyp.yaml files for the evolution to run successfully.

* Update hyp.scratch.p5.yaml

Added the anchors parameter, evolution don't start without it.

* Update hyp.scratch.p6.yaml

added the anchors parameter for the hyperparameter evolution.

* Update hyp.scratch.tiny.yaml

added the anchors parameter for the hyperparameters evolution

* Update hyp.scratch.custom.yaml

* Update hyp.scratch.tiny.yaml

* Update hyp.scratch.p5.yaml

* Update hyp.scratch.p6.yaml

* Update train.py

* Update train_aux.py
2022-07-28 22:09:52 +03:00
Kin-Yiu, Wong
a1c6e04c7c
main code
code for inference
2022-07-06 23:23:27 +08:00