Commit Graph

66 Commits

Author SHA1 Message Date
github-actions[bot]
cfbb2fd8c9 Format code 2023-09-08 16:13:28 +00:00
Steve Wang
82409e29ba Merge branch 'dev' into fix-resume 2023-09-08 20:26:15 +08:00
Steve Wang
1791d9240a Fix error on resume training: KeyError: "param 'initial_lr' is not specified in param_groups[0] when resuming an optimizer" 2023-09-08 19:17:39 +08:00
Stardust·减
67f4d08520 fix 2023-09-07 22:15:15 +08:00
Stardust-minus
82d5aaf904 Apply Code Formatter Change 2023-09-07 12:46:16 +00:00
Stardust·减
eb83b8b58b use gloo and update Dataloader 2023-09-07 20:45:51 +08:00
Stardust·减
1069066390 Update train_ms.py 2023-09-07 11:09:55 +08:00
Stardust·减
63ea12459d 更新 train_ms.py 2023-09-07 01:46:22 +08:00
Stardust-minus
c80621c7ae Apply Code Formatter Change 2023-09-06 15:00:48 +00:00
Stardust·减
57b408e8df rollback TF32 2023-09-06 23:00:33 +08:00
Stardust·减
8b61641bb1 Update train_ms.py 2023-09-06 23:00:14 +08:00
Stardust-minus
eef212fd00 Apply Code Formatter Change 2023-09-06 13:42:05 +00:00
Stardust·减
11a6e9aad7 disable TF32 2023-09-06 20:32:57 +08:00
Stardust·减
8c4316cd5e del freeze enc 2023-09-06 11:39:25 +08:00
Stardust·减
63d6f52bef Update train_ms.py 2023-09-06 10:20:42 +08:00
Stardust·减
e86c8a8991 use torchrun 2023-09-05 17:26:39 +08:00
Stardust·减
24ba0f26df Update train_ms.py 2023-09-05 11:30:28 +08:00
Stardust·减
ff8469f0f6 decrease num_worker 2023-09-05 11:26:50 +08:00
Stardust·减
5b5c79dd16 add bert 2023-09-05 10:02:50 +08:00
Stardust·减
4127f7d00a Update train_ms.py 2023-09-05 01:02:46 +08:00
Stardust·减
f83227dac5 Update train_ms.py 2023-09-05 01:00:47 +08:00
Stardust·减
4a1c09bf31 Update train_ms.py 2023-09-05 00:56:32 +08:00
Stardust·减
7761dadcc6 rollback resume 2023-09-04 22:13:57 +08:00
Stardust·减
c6f627f350 修复resume 2023-09-04 22:02:02 +08:00
Stardust·减
5914ed3efc speed optim 2023-09-04 14:27:14 +08:00
Stardust·减
d0015a0ca4 Update train_ms.py 2023-09-03 13:38:32 +08:00
Stardust·减
c18837481d fix 2023-09-03 13:14:55 +08:00
Stardust·减
95cfacc76c try 2023-09-03 12:36:53 +08:00
Stardust·减
1835ed2442 add drop_last 2023-09-03 12:35:58 +08:00
med
fe904b9f5f add --resume option 2023-09-02 01:44:45 -04:00
Stardust·减
4c35709d22 clean code 2023-09-01 10:34:14 +08:00
Stardust·减
df9438dd4a rollback 2023-08-30 14:23:45 +08:00
Stardust·减
bf2eb42a1a Update train_ms.py 2023-08-29 20:07:24 +08:00
Stardust·减
70c299faa7 修正Dur GAN 2023-08-29 20:01:48 +08:00
Stardust·减
cbd267f932 dur训练多卡 2023-08-26 23:35:33 +08:00
Stardust·减
51a03dbc17 Update train_ms.py 2023-08-23 19:00:21 +08:00
Stardust·减
1377024262 Update train_ms.py 2023-08-23 18:46:11 +08:00
Stardust·减
c9b6a9695e Update train_ms.py 2023-08-23 18:38:51 +08:00
Stardust·减
da97901831 Update train_ms.py 2023-08-23 18:19:45 +08:00
Stardust·减
81ef49f278 Update train_ms.py 2023-08-23 18:16:47 +08:00
Stardust·减
356efb724f Update train_ms.py 2023-08-23 17:21:43 +08:00
Stardust·减
e9223872c2 Update train_ms.py 2023-08-23 17:18:37 +08:00
Stardust·减
031a543150 Update train_ms.py 2023-08-23 17:17:47 +08:00
Stardust·减
1fa7a8086b Update train_ms.py 2023-08-23 17:06:45 +08:00
Stardust·减
402316fe03 Update train_ms.py 2023-08-23 17:01:53 +08:00
Stardust·减
bfc844ae02 Update train_ms.py 2023-08-23 16:53:09 +08:00
Stardust·减
e029f8491d Update train_ms.py 2023-08-23 16:49:59 +08:00
Stardust·减
432e3923f6 Update train_ms.py 2023-08-23 16:49:33 +08:00
Stardust·减
fe3768b872 Update train_ms.py 2023-08-23 16:48:31 +08:00
Stardust·减
effec21548 Update train_ms.py 2023-08-23 16:46:59 +08:00