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

..

__init__.py

missing dunder-init

2023-11-06 18:33:01 -05:00

.gitignore

Support Sample packing for phi arch (#586)

2023-09-15 15:46:54 -04:00

test_fused_llama.py

use temp_dir kwarg instead

2023-11-06 18:33:01 -05:00

test_lora_llama.py

add e2e tests for checking functionality of resume from checkpoint (#865)

2023-11-15 23:05:55 -05:00

test_mistral_samplepack.py

use temp_dir kwarg instead

2023-11-06 18:33:01 -05:00

test_mistral.py

use temp_dir kwarg instead

2023-11-06 18:33:01 -05:00

test_phi.py

use temp_dir kwarg instead

2023-11-06 18:33:01 -05:00

test_resume.py

add e2e tests for checking functionality of resume from checkpoint (#865)

2023-11-15 23:05:55 -05:00

utils.py

add e2e tests for checking functionality of resume from checkpoint (#865)

2023-11-15 23:05:55 -05:00