Commit graph

10 commits

Author SHA1 Message Date
Zangwei Zheng
3944e93065 update training config 2024-04-22 12:05:59 +08:00
Zangwei Zheng
dce2ef4a1c [exp] new config 2024-04-19 15:21:00 +08:00
Zheng Zangwei (Alex Zheng)
069ea0d687 Feat/fast bucket (#54)
* [wip] bucket

* [bug] not parallel

* update eval

* update sample.sh

* accelerate bucket build with pandarallel
2024-04-19 11:42:02 +08:00
Zheng Zangwei (Alex Zheng)
7612d22fc6 Exp/image mix (#20)
* [exp] image mixed training

* [exp] add batch info

* [exp] launch

* update num_step_per_epoch

* [feat] verify image mix training
2024-04-02 13:35:09 +08:00
Frank Lee
baf91f835e updated sampler workload and state dict (#18) 2024-04-01 21:10:51 +08:00
Frank Lee
b8b78c6733 Hotfix/sampler (#16)
* fixed variable video sampler

* polish

* polish
2024-04-01 20:37:41 +08:00
Frank Lee
a9e8220c7c fixed image size in sampler (#15)
* fixed image size in sampler

* polish
2024-04-01 11:43:56 +08:00
Frank Lee
46f9af34fc fixed multiprocessing dataloader (#14) 2024-03-31 20:59:22 +08:00
Zheng Zangwei (Alex Zheng)
f9f539f07e format and some fix (#8) 2024-03-30 13:34:19 +08:00
Hongxin Liu
5e82d1493b [feature] refactor sampler (#4)
* [feature] refactor sampler

* [feature] support sampler resuming

* [feature] support sampler resuming
2024-03-29 23:32:12 +08:00