Commit graph

39 commits

Author SHA1 Message Date
Shen-Chenhui
d8fab103f4 added disc 2024-05-02 09:48:30 +00:00
Shen-Chenhui
bcb997502e enable disc 2024-05-02 17:28:45 +08:00
Shen-Chenhui
b35b933841 add disc 2024-05-02 16:32:19 +08:00
zhengzangw
02904b4f29 updated 2024-05-02 03:46:01 +00:00
zhengzangw
ace37cf7c7 refactor 2024-04-30 08:13:20 +00:00
zhengzangw
41e276f5ef updated 2024-04-30 06:19:40 +00:00
zhengzangw
b2f302b379 [wip] try vae 2024-04-29 17:02:22 +00:00
Shen-Chenhui
c584743d50 debug 2024-04-29 20:10:15 +08:00
Shen-Chenhui
c631358bbf add z nll loss 2024-04-29 18:09:42 +08:00
Shen-Chenhui
f81d8648cf move out losses to different file 2024-04-29 17:53:03 +08:00
Shen-Chenhui
4d69671663 clean up code 2024-04-29 17:18:45 +08:00
Shen-Chenhui
a4025b0ea7 magvit v2 enc dec arc 2024-04-09 17:49:01 +08:00
Shen-Chenhui
ae6bffb8c5 restore train vae script 2024-04-09 09:34:59 +08:00
Shen-Chenhui
ce0b1928e6 debug 2024-04-08 17:41:08 +08:00
Shen-Chenhui
04a0e4769b debug 2024-04-08 17:33:54 +08:00
Shen-Chenhui
fb64216b45 debug 2024-04-08 17:32:30 +08:00
Shen-Chenhui
25707a9d7d debug 2024-04-08 17:28:31 +08:00
Shen-Chenhui
9c8d084ec5 debug 2024-04-08 17:26:36 +08:00
Shen-Chenhui
eae30f9f89 debug 2024-04-08 17:23:53 +08:00
Shen-Chenhui
ab1970ca24 debug 2024-04-08 16:05:15 +08:00
Shen-Chenhui
0a5c767fc7 debug 2024-04-08 15:54:03 +08:00
Shen-Chenhui
b2148d5505 debug 2024-04-08 15:52:25 +08:00
Shen-Chenhui
7e3951300f debug 2024-04-08 15:43:06 +08:00
Shen-Chenhui
fa0ca3983e debug inference code 2024-04-08 15:40:14 +08:00
Shen-Chenhui
9091419b28 save vae to model dir 2024-04-05 16:07:23 +08:00
Shen-Chenhui
dfdf8db491 add inference config
add readme
complete inference script
TODO: DEBUG inference
2024-04-04 14:11:25 +08:00
Shen-Chenhui
ebb3dc4d59 enable wandb 2024-04-02 11:47:56 +08:00
Shen-Chenhui
562a966a77 debug 2024-04-01 16:14:11 +08:00
Shen-Chenhui
3d6d499543 add perceptual loss 2024-04-01 13:57:25 +08:00
Shen-Chenhui
d1e57cbfd5 debug 2024-03-29 10:30:56 +08:00
Shen-Chenhui
4795fd5354 debug 2024-03-28 23:27:26 +08:00
Shen-Chenhui
3692723b4b debug 2024-03-28 18:30:36 +08:00
shenchenhui
41f7b9b01e debug 2024-03-28 18:17:42 +08:00
shenchenhui
7f105cbdfa debug 2024-03-28 18:02:30 +08:00
shenchenhui
716b3f3774 debug 2024-03-28 17:45:50 +08:00
shenchenhui
42101a5eee debug 2024-03-28 17:38:48 +08:00
shenchenhui
5935c5b10b debug 2024-03-28 17:36:31 +08:00
shenchenhui
3d9addba17 debug 2024-03-28 16:09:43 +08:00
shenchenhui
a5917c6a3e added vae3d training code 2024-03-28 15:12:20 +08:00