Commit graph

54 commits

Author SHA1 Message Date
Frank Lee
a887f54711 added sp for stdit3 (#131) 2024-06-14 11:10:43 +08:00
zhengzangw
68778cda9f update train config 2024-06-07 05:49:34 +00:00
zhengzangw
ce601066c5 [stage2] update config 2024-06-05 15:49:51 +00:00
Frank Lee
ea96cdb5e4 added warmup lr scheduler (#121) 2024-06-05 10:12:52 +08:00
zhengzangw
fd51586704 [fix] start_from_scratch 2024-05-31 06:46:30 +00:00
zhengzangw
ee1354a7e0 [fix] release av threads 2024-05-30 08:49:39 +00:00
zhengzangw
8a686660bc read video 2024-05-30 08:09:03 +00:00
Frank Lee
2e1e26a2e4 added timer for benchmarking (#114) 2024-05-29 14:43:15 +08:00
zhengzangw
fe60a93240 update feat train 2024-05-21 09:55:26 +00:00
zhengzangw
a50bec768f [feat] complete training with feature 2024-05-21 07:20:14 +00:00
zhengzangw
066b0c9bb3 [wip] feat training 2024-05-21 05:45:06 +00:00
zhengzangw
80f9ecbde3 [feat] generate feature 2024-05-21 04:05:02 +00:00
zhengzangw
bdc67a3933 [wip] debug extract feat 2024-05-20 09:40:28 +00:00
zhengzangw
5b9e753039 [feat] unify dataloader api 2024-05-20 08:40:45 +00:00
zhengzangw
9fd4554f55 [fix] vae dtype 2024-05-16 08:50:24 +00:00
Frank Lee
d1b73a7c9f handled erroneous data (#101) 2024-05-14 14:27:46 +08:00
zhengzangw
3a01cb440d [feat] inference for rf 2024-05-13 11:39:16 +00:00
zhengzangw
6194da6504 [fix] update inference and datautil 2024-05-13 06:33:12 +00:00
zhengzangw
74dfaa879c [fix] ema ckpt save 2024-05-12 05:15:06 +00:00
zhengzangw
e57fd9a62c [wip] inference refactor 2024-05-10 13:26:14 +00:00
zhengzangw
2a55d672a7 [feat] update logging and bucket 2024-05-09 08:07:56 +00:00
zhengzangw
e9db7b3af1 update training script 2024-05-09 05:53:19 +00:00
zhengzangw
31228fe9c8 [refactor] fix train warnings 2024-05-09 03:10:26 +00:00
zhengzangw
322e566aed update 2024-05-09 03:10:26 +00:00
Zangwei Zheng
eb5ff82fc5 quick update 2024-05-09 11:09:52 +08:00
Shen Chenhui
dc7b7fd64a fix bug (#95)
Co-authored-by: Shen-Chenhui <shen_chenhui@u.nus.edu>
2024-05-08 17:39:21 +08:00
Zheng Zangwei (Alex Zheng)
f047d5786a [refactor] clean train.py code (#94) 2024-05-08 16:07:57 +08:00
Frank Lee
fe19568a49 fixed race condition in exp workspace (#91) 2024-05-07 17:17:43 +08:00
Zangwei Zheng
8564074229 [exp] 2024-05-07 15:41:52 +08:00
Zangwei Zheng
917643d5bd [wip] 2024-05-07 14:51:11 +08:00
FrankLeeeee
ec88d2d54e resolved conflicts 2024-05-07 03:08:41 +00:00
Frank Lee
d599fd75b3 adapted pretrained model to training (#371) 2024-05-07 10:37:06 +08:00
Zangwei Zheng
eefb351333 [exp] update config bucket 2024-05-06 14:32:28 +08:00
Zangwei Zheng
c23fe85a70 [exp] 2024-05-04 16:16:06 +08:00
Zangwei Zheng
fd24a8c4f9 update adaptation 2024-05-04 11:51:31 +08:00
zhengzangw
1e702a3088 fix start from scratch 2024-04-23 11:34:35 +00:00
Zangwei Zheng
e009e89e1a add start from scratch 2024-04-23 19:24:22 +08:00
Zangwei Zheng
f6574f1509 hotfix exp 2024-04-19 18:15:06 +08:00
Zangwei Zheng
15f9d702ed [exp] update 2024-04-19 17:43:44 +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
Frank Lee
79dabf8bdf added seed to dataloader args (#52) 2024-04-16 17:45:53 +08:00
Frank Lee
db94454645 Feature/timeout (#50)
* added large nccl timeout

* polish

* polish
2024-04-15 23:55:58 +08:00
Zangwei Zheng
e88185fb9f update notebook 2024-04-14 02:02:59 +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
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
Zangwei Zheng
cba02e8a58 [feat] dynamic for video (base_size not completed) 2024-03-28 17:58:36 +08:00
Zangwei Zheng
94e686177e [wip] multi ar 2024-03-27 23:04:12 +08:00
Zangwei Zheng
7392d2e551 [feat] multiple frames with 360p 2024-03-27 00:24:46 +08:00
Zangwei Zheng
552b7e8f79 register dataset 2024-03-26 17:02:41 +08:00