Commit Graph

51 Commits (e4e9153a1f8730ad35000f5232fd261ee29f613f)

Author SHA1 Message Date
jingyaogong 568a863520 [fix] bugs 2026-04-21 13:03:34 +08:00
jingyaogong e612cc26e8 [fix] lora compile 2026-04-10 00:01:21 +08:00
jingyaogong 862dd151f2 [fix] lora moe 2026-04-09 16:36:48 +08:00
jingyaogong b703844a60 [update] add a comment 2026-04-09 15:09:58 +08:00
jingyaogong 2c24568176 [update] change default seq_len 2026-03-26 10:09:06 +08:00
jingyaogong b24fcdc1f9 [update] minimind-3 2026-03-25 23:57:45 +08:00
readlnh c88c497795 [fix] align log/save last-step check and ETA with 1-indexed step 2026-03-24 02:01:40 +01:00
readlnh 9fe4343b7b [fix] gradient accumulation step alignment 2026-03-24 01:45:04 +01:00
jingyaogong 69180447af [update] save interval 2026-01-30 20:30:50 +08:00
jingyaogong 3d2f03cfff [fix] data skip 2026-01-18 16:56:29 +08:00
jingyaogong f0aa3ba52e [update] shuffle data 2026-01-18 16:39:34 +08:00
jingyaogong b189f0046a [update] align loss 2026-01-15 00:56:32 +08:00
jingyaogong caf76fb884 [fix] compile unpack 2026-01-14 20:13:32 +08:00
jingyaogong 6ef2105e34 [feat] add compile 2026-01-14 14:42:30 +08:00
jingyaogong 379983849e [update] params log 2026-01-07 23:08:45 +08:00
jingyaogong 6bea38c741 [update] rename train tokenizer 2026-01-06 01:17:33 +08:00
jingyaogong cb6be70365 [update] rename reason 2026-01-05 23:12:29 +08:00
jingyaogong dbfeff23e3 [fix] dist cleanup 2026-01-02 22:25:55 +08:00
jingyaogong 3c56aca1c1 [update] aux loss 2026-01-01 22:41:46 +08:00
jingyaogong 5b87b9ad8a [fix] experts unused 2025-12-31 21:47:04 +08:00
jingyaogong e82af4bf8f [fix] layers set 8 2025-12-31 21:06:37 +08:00
jingyaogong 1c333636c5 [fix] moe unused 2025-12-31 21:00:06 +08:00
jingyaogong 73a2bcc412 [feat] get params 2025-12-31 20:46:59 +08:00
jingyaogong 759036b822 [feat] get params 2025-12-31 20:44:34 +08:00
jingyaogong 6ee867fffc [feat] update lr 2025-12-31 10:27:09 +08:00
jingyaogong fc2eeff509 [feat] remove empty_cache 2025-12-27 07:14:36 +08:00
jingyaogong 658ddcae3a [feat] explicit left padding 2025-12-23 18:59:48 +08:00
jingyaogong dacd4e5121 [feat] adjust seq length 2025-12-14 20:41:58 +08:00
jingyaogong 37300329c5 [fix] dtype & lr 2025-12-09 13:01:38 +08:00
dyhuachi 8fcc81696d [fix] Refactor get_lr function to include min_lr calculation
这里的退火算法会让参数里的lr的起始值变成原来lr的1.1倍,作出如下修改
2025-12-06 17:09:51 +08:00
jingyaogong 95bcb7ba19 [fix] cuda memory #559 2025-12-01 16:17:43 +08:00
jingyaogong cafc3b22df [feat] release memory 2025-11-27 19:39:49 +08:00
jingyaogong 3382c90246 [fix] ppo mask 2025-11-19 23:39:02 +08:00
jingyaogong c8727733f3 [fix] prompt length calculate 2025-11-15 18:25:37 +08:00
jingyaogong 86d5cc70ec [feat] clear cache 2025-11-06 13:12:28 +08:00
jingyaogong 7c3d660eec [feat] update import 2025-10-31 23:45:55 +08:00
jingyaogong ccab16a9b9 [feat] add args 2025-10-30 10:05:12 +08:00
jingyaogong 230b29b94a [fix] model device 2025-10-29 10:36:28 +08:00
jingyaogong e29f695998 [feat] update trainer 2025-10-29 00:52:37 +08:00
jingyaogong cd576b58e8 [feat] update trainer 2025-10-28 23:30:10 +08:00
jingyaogong 43726a7c34 [feat] pause-training 2025-10-26 18:49:52 +08:00
jingyaogong 8d1778a186 [feat] shuffle data 2025-10-23 20:13:28 +08:00
jingyaogong 3658bdb20d [fix] loss-issues-430 2025-10-23 19:08:42 +08:00
jingyaogong fa935a496e [fix] issue-431 2025-10-23 18:56:30 +08:00
jingyaogong 1e4f464553 [fix] sampler-ddp 2025-10-23 15:03:19 +08:00
jingyaogong faec18117e [fix] graph-oom & ddp-pos_cis 2025-10-23 14:22:13 +08:00
jingyaogong 249fb5b7ef [feat] minimind-2510 2025-10-21 21:19:47 +08:00
jingyaogong 71baefa98a fix bugs 2025-04-29 20:45:39 +08:00
jingyaogong 5343923e0f update lora 2025-04-27 15:45:06 +08:00
jingyaogong 7d81a0a6f0 fix bugs 2025-04-27 09:56:49 +08:00