43 Commits

Author	SHA1	Message	Date
hiyouga	5111cac6f8	support report custom args	2024-12-21 21:42:45 +00:00
hoshi-hiyouga	947e22a4a3	Merge pull request #6401 from Zeyi-Lin/hiyouga/swanlab ... feat: add swanlab for experiment tracking and visualization.	2024-12-21 14:09:33 +08:00
ZeYi Lin	3a7ea2048a	fix: by hiyouga suggestion	2024-12-20 16:43:03 +08:00
hiyouga	c7cedc7569	support disable shuffling	2024-12-19 08:53:21 +00:00
hiyouga	2811814fc4	fix mrope	2024-12-12 15:08:17 +00:00
hiyouga	1324d158f9	support batch infer in vllm	2024-12-04 13:50:00 +00:00
hoshi-hiyouga	bd639a137e	Merge pull request #6078 from wtmlon/support-efficient-tokens-calculation ... support effective tokens calculation on sft/dpo	2024-11-20 13:43:15 +08:00
Ting	40627c601e	code refactor	2024-11-19 20:33:18 +08:00
Ting	f566ecc8d1	update	2024-11-19 19:12:10 +08:00
Ting	ef6e14550d	update	2024-11-19 19:10:07 +08:00
Ting	b9f00286d8	support efficient tokens calculation on sft/dpo	2024-11-19 17:15:47 +08:00
hoshi-hiyouga	dc82821872	fix #6050	2024-11-16 16:11:16 +08:00
hiyouga	4270f7dfb9	fix dpo metrics	2024-11-02 20:59:01 +08:00
hiyouga	30567a1487	fix incorrect loss value for vlms	2024-10-30 08:56:46 +00:00
hiyouga	ae045c884f	fix #5747	2024-10-29 10:47:04 +00:00
hiyouga	21db8ed2f4	use pre-commit	2024-10-29 09:07:46 +00:00
hiyouga	54c6905937	add docstrings, refactor logger	2024-09-08 00:56:56 +08:00
hiyouga	dabad5570b	update get template	2024-09-04 22:36:20 +08:00
hiyouga	47ea97fb1b	lazy image load	2024-09-04 02:27:08 +08:00
hiyouga	8e49940746	add rlhf-v dataset	2024-09-01 22:57:41 +08:00
moontidef	40908a36fa	fix: rename optimzer to optimizer	2024-08-07 10:05:01 +08:00
hiyouga	29ebcd75d5	fix up	2024-07-15 01:04:56 +08:00
hoshi-hiyouga	15b399a82f	Merge pull request #4691 from codemayq/feature-suppot-eval-dataset ... add eval dataset support	2024-07-15 01:00:34 +08:00
hiyouga	2f09520c0d	fix #4742	2024-07-09 23:24:24 +08:00
codingma	76f3bbcfc0	1. add custom eval dataset support ... 2. merge load dataset and split dataset function	2024-07-05 15:52:10 +08:00
hiyouga	8baf3b22b0	refactor pissa, improve llamaboard	2024-06-28 01:04:24 +08:00
hiyouga	095fab58d3	tiny fix about badam	2024-06-25 01:54:53 +08:00
Jonery	5c2ff1b749	Cleaner integration.	2024-06-19 12:29:40 +08:00
Jonery	0f72aac8c9	Support distributed BAdam.	2024-06-18 12:27:47 +08:00
hiyouga	8c1046d78a	support pissa	2024-06-16 01:08:12 +08:00
hiyouga	38b6b0f52e	tiny fix	2024-06-16 01:06:41 +08:00
hiyouga	d87108daa6	add license	2024-06-15 17:54:33 +08:00
hiyouga	cf9f2d6c42	fix #4209 ... DeepSpeed ZeRO3 has inflight param error when calling model.eval()	2024-06-13 02:25:50 +08:00
hiyouga	f9e818d79c	fix #4120	2024-06-07 04:18:05 +08:00
hiyouga	74f96efef9	rename files	2024-06-07 00:09:06 +08:00
hiyouga	fad2591e31	update trainers	2024-06-06 18:45:49 +08:00
hiyouga	67fe822324	fix #4090	2024-06-06 00:50:32 +08:00
hiyouga	f9a206509e	remove gc warnings in DPO&KTO	2024-06-03 22:53:54 +08:00
hiyouga	7c8e01bb74	update dpo, kto trainer	2024-05-29 00:14:29 +08:00
hiyouga	cb63b32986	support SimPO #3900	2024-05-26 23:46:33 +08:00
hiyouga	3a023bca2a	refactor data preprocessing, fix mllm rlhf	2024-05-24 04:08:25 +08:00
hiyouga	c450ee87a3	improve KTO impl., replace datasets	2024-05-18 03:44:56 +08:00
hiyouga	308edbc426	rename package	2024-05-16 18:39:08 +08:00