3 Commits

Author	SHA1	Message	Date
Wing Lian	b3a61e8ce2	add e2e tests for checking functionality of resume from checkpoint (#865) ... * use tensorboard to see if resume from checkpoint works * make sure e2e test is either fp16 or bf16 * set max_steps and save limit so we have the checkpoint when testing resuming * fix test parameters	2023-11-15 23:05:55 -05:00
Wing Lian	6dc68a653f	use temp_dir kwarg instead	2023-11-06 18:33:01 -05:00
Wing Lian	c74f045ba7	chore: lint	2023-11-06 18:33:01 -05:00