1612 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
Jake Poznanski	43ccd82609	One more train config with rotation augments	2025-09-10 16:06:04 +00:00
Jake Poznanski	3eaa584ed5	Fixing data config	2025-09-09 15:22:04 +00:00
Jake Poznanski	077e3eea7f	Preemptible mix	2025-09-09 14:55:01 +00:00
Jake Poznanski	54cd5a3438	Going to train on the new transcripts data	2025-09-08 22:30:40 +00:00
Jake Poznanski	592a669e1f	Arxiv downloader	2025-09-08 20:44:54 +00:00
Jake Poznanski	607e251530	Mine tables stuff	2025-09-08 20:22:18 +00:00
Jake Poznanski	3e9477db98	Aiming to get some more table data	2025-09-08 20:05:09 +00:00
Jake Poznanski	ba8b3824bf	Adding some rotation augmentation to the post training step	2025-09-08 18:54:53 +00:00
Jake Poznanski	a957ab2aaf	Adding an adjustment to how blank pages test is run, skipping image tags	2025-09-08 17:18:51 +00:00
Jake Poznanski	c0a1e70440	Adding filter	2025-09-08 17:02:09 +00:00
Jake Poznanski	aad8729ad7	prepare checkpint from main	2025-09-08 17:01:58 +00:00
Jake Poznanski	0f46fa0988	Adding num_generations	2025-09-04 18:44:23 +00:00
Jake Poznanski	ef09c73bf2	Fixing up some rewards stuff	2025-09-04 17:34:53 +00:00
Jake Poznanski	ede0dc51b1	Adding drop last to prevent any weirdnesses	2025-09-04 16:50:08 +00:00
Jake Poznanski	14a882db9a	Fixing to new version, adjusting scale rewards stuff	2025-09-03 22:43:35 +00:00
Jake Poznanski	2fd4ae8489	Adding some more options to play with	2025-09-03 22:29:23 +00:00
Jake Poznanski	755c221024	Trying some more things	2025-09-03 22:11:16 +00:00
Jake Poznanski	0a9c8f3e96	Adding warmup steps param	2025-09-03 21:33:18 +00:00
Jake Poznanski	a41d04660a	Cleaning script	2025-09-03 21:31:21 +00:00
Jake Poznanski	e6cff25b6b	Cleanup stuff	2025-09-03 20:34:12 +00:00
Jake Poznanski	bade86fe91	Cleaned up things	2025-09-03 20:23:01 +00:00
Jake Poznanski	b689a8e5f8	Giving more memory buffer	2025-09-03 19:56:53 +00:00
Jake Poznanski	7346d12322	Better cleaning, augusta version	2025-09-03 18:47:02 +00:00
Jake Poznanski	f20f1a0b54	Doing some cleaning	2025-09-03 18:41:36 +00:00
Jake Poznanski	94d19c51c6	Cleaning up scripts, multi gpu trainer more flexible	2025-09-03 18:25:10 +00:00
Jake Poznanski	c612293a59	Remove device map auto	2025-09-03 18:04:42 +00:00
Jake Poznanski	1fb49cefc1	Working on multi gpu trainer	2025-09-03 17:25:14 +00:00
Jake Poznanski	00f51fb2c7	Fixing bug with multi epoch training	2025-09-02 21:03:00 +00:00
Jake Poznanski	72fcfafde7	Fixed up national archives script	2025-08-29 16:36:18 +00:00
Jake Poznanski	7e09a02f7c	Better NA extractor	2025-08-28 20:48:51 +00:00
Jake Poznanski	22626a512c	Describing national archives project	2025-08-28 18:44:24 +00:00
Jake Poznanski	f426826850	Clean downloads	2025-08-28 18:03:52 +00:00
Jake Poznanski	3d3d184f25	Fixes	2025-08-28 17:56:27 +00:00
Jake Poznanski	ed3820c0c7	LOC downloader	2025-08-28 17:38:20 +00:00
Jake Poznanski	6123d4452b	Reorganizing files	2025-08-28 17:04:26 +00:00
Jake Poznanski	0710debf75	Cleaner front matter reward	2025-08-27 19:49:42 +00:00
Jake Poznanski	09036b07d9	Verifying bench loading	2025-08-27 19:00:29 +00:00
Jake Poznanski	14f19e5d58	Some cost tracking	2025-08-27 18:53:42 +00:00
Jake Poznanski	d70208d98a	Moving test code around, adding format reward since some runs stop outputting the front matter thing in grpo training	2025-08-27 18:22:05 +00:00
Jake Poznanski	8383865392	Fixing up subscripts and superscripts in synth data	2025-08-27 18:15:36 +00:00
Jake Poznanski	cf479d1d29	Fixing reward weights stuff	2025-08-26 22:04:45 +00:00
Jake Poznanski	c86b413d3e	Adding additional rewards and weights	2025-08-26 21:49:35 +00:00
Jake Poznanski	9c520498dd	Adding some losses	2025-08-26 21:18:24 +00:00
Jake Poznanski	3be381b375	Adding some params	2025-08-26 20:46:06 +00:00
Jake Poznanski	4b3660debd	Quieting some messages	2025-08-26 20:15:39 +00:00
Jake Poznanski	8327da2415	Fix	2025-08-26 20:06:52 +00:00
Jake Poznanski	98f4d62d1e	Fixing data check	2025-08-26 19:51:54 +00:00
Jake Poznanski	3433c8f5f2	Adding more cmd line args	2025-08-26 18:28:41 +00:00
Jake Poznanski	9671f6847c	Parallel rewards	2025-08-26 18:16:21 +00:00
Jake Poznanski	2588f911dc	Docker ignore	2025-08-26 17:52:30 +00:00