Commit Graph

3 Commits (8352951f3de379d7e4b21d85e42da7020ff4c0c1)

Author SHA1 Message Date
fam_taro 4ce7be17c9
[Enhance] Enable to toggle whether Gem Pooling is trainable or not. (#1246)
* Enable to toggle whether Gem Pooling is trainable or not.

* Add test case whether Gem Pooling is trainable or not.

* Enable to toggle whether Gem Pooling is trainable or not by requires_grad

---------

Co-authored-by: Yusuke Fujimoto <yusuke.fujimoto@rist.co.jp>
2023-02-09 11:27:05 +08:00
zzc98 9eb6fc4368
[Feature] Add reduction for neck (#978)
* feat: add reduction for neck

* feat: add reduction for neck

* feat: add reduction for neck

* feat:add linear reduction neck

* feat: add reduction neck

* mod out of linearReduction as tuple

* fix typo

* fix unit tests

* fix unit tests

Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
2022-11-04 15:44:37 +08:00
Ma Zerun b8b31e9343
[Improve] Multiple fixes. (#957)
* Rename some files in unit tests.

* Rename `pipelines` folder to `transforms`.

* Fix version requirement range.

* Remove `init_random_seed` and `set_random_seed`.

* Fix unit tests

* Rename `convert_models` to `model_converters`.
2022-08-09 15:58:08 +08:00