diff --git a/tests/save_dir/_ckpt_epoch_1.ckpt b/tests/save_dir/_ckpt_epoch_1.ckpt
new file mode 100644
index 00000000..84a29397
Binary files /dev/null and b/tests/save_dir/_ckpt_epoch_1.ckpt differ
diff --git a/tests/save_dir/_ckpt_epoch_2.ckpt b/tests/save_dir/_ckpt_epoch_2.ckpt
new file mode 100644
index 00000000..7c32d2d2
Binary files /dev/null and b/tests/save_dir/_ckpt_epoch_2.ckpt differ