mmpretrain

Author	SHA1	Message	Date
Yixiao Fang	63d9f27fde	[Refactor] Add necks, heads and losses for the self-supervised task. (#1376 ) * add necks * refactor linear neck * rename simmim neck * add heads * add losses * fix * add unittest * update * update cae * remove mim head * update config	2023-02-28 10:05:00 +08:00
mzr1996	0979e78573	Rename the package name to `mmpretrain`.	2023-02-17 15:20:55 +08:00
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