search

sweetcocoa / pop2piano

Official Repo of the paper "Pop2Piano : Pop Audio-based Piano Cover Generation"
448 stars 46 forks source link

global_step KeyError in Colab #18

Closed Branchverse closed 1 year ago

Branchverse commented 1 year ago

when running this part of code in the Colab I get the KeyError:

wrapper = wrapper.load_from_checkpoint("model-1999-val_0.67311615.ckpt", config=config).to(device)
[/usr/local/lib/python3.10/dist-packages/pytorch_lightning/utilities/migration/migration.py](https://localhost:8080/#) in _migrate_loop_global_step_to_progress_tracking(checkpoint)
     89     PR: #13645, #11805
     90     """
---> 91     global_step = checkpoint["global_step"]
     92     checkpoint.setdefault("loops", {"fit_loop": _get_fit_loop_initial_state_1_6_0()})
     93     checkpoint["loops"].setdefault("fit_loop", _get_fit_loop_initial_state_1_6_0())

KeyError: 'global_step'