Commit Graph

371 Commits (1398e4200e19386894829a4418c03a491b434a9f)

Author SHA1 Message Date
fanqiNO1 a5db5bedb9
[Fix] Fix CI for torch2.1.0 (#1418) 2023-11-20 19:31:14 +08:00
Shu Liqiang b8a31671a4
[Feature] Runner supports setting the number of iterations for per epoch (#1292) 2023-10-08 16:43:44 +08:00
Mashiro 95d875832a
[Enhance] Support for installing minimal runtime dependencies (#1362) 2023-10-08 16:38:40 +08:00
hiyyg eb5834fa66
[Enhance] metainfo of dataset can be a generic dict-like Mapping (#1378) 2023-10-08 14:25:26 +08:00
6V e9e08dbb65
[Enhance] Add unit tests for autocast with Ascend device (#1363) 2023-09-27 10:20:13 +08:00
takuoko d617bcafdd
[Feature] Support Adafactor Optimizer (#1361) 2023-09-21 16:30:24 +08:00
Mashiro 53474ef1ba
[Fix] Fix get class attribute from a string (#1345) 2023-09-18 11:50:41 +08:00
takuoko e91bfa4593
[Feature] Support bitsandbytes (#1357) 2023-09-15 22:56:11 +08:00
Range King 9a19f8bc89
[Feature] Add AimVisBackend (#1347) 2023-09-09 11:39:48 +08:00
Range King 273fb2b333
[Feature] Add DVCLiveVisBackend (#1336) 2023-09-08 17:22:23 +08:00
Zeyuan ccd17571ce
[Feature] Implement gradient checkpointing (#1319) 2023-09-04 23:29:24 +08:00
Zaida Zhou 9aa883a24c
Fix the type check of tasks in progress bar (#1340) 2023-09-04 19:27:58 +08:00
Kevin Wang 8a7e80e9e0
[Feature] Support using other file handlers (#1188) 2023-08-30 20:37:39 +08:00
王永韬 0939d95c93
[Feature ] Add progressbar rich (#1157) 2023-08-30 20:10:07 +08:00
Mashiro 170758aefe
[Fix] Fix get optimizer_cls (#1324) 2023-08-28 16:15:00 +08:00
Theodore 43e308caaf
[Feature] Add NeptuneVisBackend (#1311) 2023-08-17 23:29:58 +08:00
Mashiro 45a3d310be
[Fix] Skip adding `vis_backend` when `save_dir` is not set (#1289) 2023-08-03 20:27:45 +08:00
Xinyu Yang 4fdab5e9cb
[Fix] Fix Visualizer that built vis_backends will not be used when save_dir is None (#1275) 2023-07-31 15:29:44 +08:00
Zaida Zhou 2df93eb51f
Add the loop stage in message_hub (#1277) 2023-07-31 14:22:49 +08:00
Mashiro 237aee3866
[Enhance] Enhance error information in build function (#1088) 2023-07-28 11:21:09 +08:00
youkaichao ee742da254
[Enhance] Use graph transform to deal with more general cases for efficient_conv_bn_eval (#1259) 2023-07-26 17:52:57 +08:00
vugia truong c8a1264568
[Feature] Compare the difference of two configs (#1260) 2023-07-26 15:48:59 +08:00
KerwinKai 68360e7ce8
[Feature] Add parameter `save_begin` for CheckpointHook (#1271) 2023-07-25 19:21:21 +08:00
Mashiro 3871881ef6
[Enhance] Support skipping initialization in `BaseModule` (#1263) 2023-07-25 12:56:42 +08:00
youkaichao 66d828d8d3
[Enhancement] Rename fast_conv_bn_eval to efficient_conv_bn_eval (#1251) 2023-07-15 22:13:17 +08:00
i-aki-y 276c614bf5
[Fix] Fix scalar check in RuntimeInfoHook (#1250) 2023-07-15 22:03:43 +08:00
youkaichao 40e49ff747
[Feature] Enable fast conv bn eval (#1202) 2023-07-14 18:21:55 +08:00
Qingyun de81d29abd
[Fix] Fix the logic of setting lazy_import (#1239) 2023-07-11 22:32:20 +08:00
Mashiro 955b5712c4
[Fix] Fix pickle the Python style config (#1241) 2023-07-11 20:37:48 +08:00
Mashiro 8d4bac2616
[Enhance] Enhance config (#1232) 2023-07-05 23:27:08 +08:00
Mashiro 317d8f3178
[Fix] Remove auto find_ignored_modules (#1231) 2023-07-03 19:54:54 +08:00
Mashiro d5a46d4144
[Fix] Implement `copy` and `__copy__` for `ConfigDict` (#1230) 2023-07-03 15:11:10 +08:00
Mashiro 20d477dae1
[Enhancement] Only recognize modules `mixed gradients` as `ignored_modules` (#1226) 2023-07-01 22:50:36 +08:00
Mashiro b638d3b1fe
[Fix] Fix new config (#1227) 2023-07-01 22:35:11 +08:00
Mashiro 1480261e8f
[Enhance] Config adds copy method (#1224) 2023-06-30 17:11:16 +08:00
Mashiro f930b9fe53
Fix docstring format (#1223) 2023-06-30 10:39:19 +08:00
Mashiro 399f76ffa8 [Experimental] Add support for FSDP (#1213) 2023-06-29 15:19:33 +08:00
Mashiro 478c952a6d
[Refacor] Replace 'if base' with 'with read_base' context manager (#1207) 2023-06-25 13:53:19 +08:00
Mashiro 04b0ffee76
[Fix] Fix ut error in docker (#1204) 2023-06-16 22:05:08 +08:00
Mashiro 6ece63ed35
[Feature] Support Pure Python style Configuration File (#1071) 2023-06-16 12:52:07 +08:00
Akide Liu 94e7a3bb57
[Enhance] Learning rate in log can show the base learning rate of optimizer (#1019) 2023-06-08 19:51:15 +08:00
Zaida Zhou 19aa1eb780
[Fix] Save checkpoint again to update best_ckpt of ckpt (#1168) 2023-06-02 14:42:56 +08:00
i-aki-y 6df9621a06
[Feature] Add support for full wandb's define_metric arguments (#1099) 2023-06-01 21:50:29 +08:00
vugia truong 68414516aa
[Feature] Add vis backend for clearml (#878) (#1091) 2023-06-01 17:41:34 +08:00
Zaida Zhou 193b7fdfcc
[Refactor] Let unit tests not affect each other (#1169) 2023-05-27 22:36:04 +08:00
Mashiro 5d4e72144a
[Fix] Fix `ProfileHook` can not profile performance in ddp-training (#1140) 2023-05-26 10:55:15 +08:00
Mashiro 2085046d22
[Fix] The ann_file and data_root of BaseDataset can be None (#850) 2023-05-04 22:22:52 +08:00
Mashiro 3715fea15b
[Refactor] Refactor the unit tests of SyncBuffersHook (#813) 2023-04-28 17:32:30 +08:00
Mashiro 298a4b1e49
[Fix] Fix build unnecessary loop during train/test/val (#1107)
* [Fix] Fix build unnecessary loop during train/test/val

* move unit test to runner

* Update unit test

* Fix unit test

* check train_loop is None

* update comment

* replace(type(None)) with is not None
2023-04-27 19:20:35 +08:00
Ma Zerun 49b27dd83f
[Imporve] Support `_load_state_dict_post_hooks` in `load_state_dict`. (#1103)
* [Imporve] Support `_load_state_dict_post_hooks` in `load_state_dict`.

* Update

* Add unit test
2023-04-26 16:48:57 +08:00