merlot_reserve
merlot_reserve copied to clipboard
Failed to restore checkpoint
Hi, I installed the package following your guidance. However, when I ran the demo_video.py, it raised a ValueError: Unpack failed: incomplete input when doing state=checkpoints.restore_checkpoint(ckpt_dir_path, target_state, step=step, prefix='ckpt_', parallel=True) in line 125 of mreserve/checkpoint.py. What should I do?
My flax version is 0.3.4 and the large_resadapt checkpoint is auto-downloaded.