Spijkervet / CLMR

Official PyTorch implementation of Contrastive Learning of Musical Representations

Home Page:https://arxiv.org/abs/2103.09410

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

PyTorch Lightning version

EmreOzkose opened this issue · comments

Hi,

When I followed these steps to test pre-trained model, I got this error:

(clmr) yunusemre.ozkose@server:/CLMR$ python main.py --dataset audio --dataset_dir /path/to/test_audio_dataset_dir
Global seed set to 42
Traceback (most recent call last):
  File "main.py", line 129, in <module>
    module = module.load_from_checkpoint(
  File "/miniconda3/envs/cmlr/lib/python3.8/site-packages/pytorch_lightning/core/saving.py", line 137, in load_from_checkpoint
    return _load_from_checkpoint(
  File "/miniconda3/envs/cmlr/lib/python3.8/site-packages/pytorch_lightning/core/saving.py", line 205, in _load_from_checkpoint
    return _load_state(cls, checkpoint, strict=strict, **kwargs)
  File "/miniconda3/envs/cmlr/lib/python3.8/site-packages/pytorch_lightning/core/saving.py", line 250, in _load_state
    obj = cls(**_cls_kwargs)
TypeError: __init__() missing 1 required positional argument: 'args'

There was a related bug in Lightning that seems to be solved. Could you state your PyTorch Lightning version?

have you solved this issue? I'm having trouble with this same issue.

Hi @ee-jw , actually I even don't remember where/when I used this model, sorry.