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 |
|