--model 'transformer_r10.4.1_5khz_v4.3' has error "UnboundLocalError: local variable 'model' referenced before assignment "

[kerenzhou062]

when running deepmod2 detect --model 'transformer_r10.4.1_5khz_v4.3', it throws error like:

Traceback (most recent call last): File "/hpcf/authorized_apps/rhel8_apps/deepmod2/install/deepmod2-env2/lib/python3.10/multiprocessing/process.py", line 314, in _bootstrap self.run() File "/hpcf/authorized_apps/rhel8_apps/deepmod2/install/deepmod2-env2/lib/python3.10/multiprocessing/process.py", line 108, in run self._target(*self._args, **self._kwargs) File "/research/rgs01/applications/hpcf/authorized_apps/rhel8_apps/deepmod2/vendor/DeepMod2/src/detect.py", line 419, in process model, model_config=get_model(params) File "/research/rgs01/applications/hpcf/authorized_apps/rhel8_apps/deepmod2/vendor/DeepMod2/src/utils.py", line 108, in get_model model.load_state_dict(checkpoint['model_state_dict']) UnboundLocalError: local variable 'model' referenced before assignment`

[umahsn]

Hi,

Are you using the latest commit? This bug was fixed in commit 2f7c90daf463a289466c40ff1dac606503cf5f77, so if you use the latest version you should not have this error. Please use git pull command within DeepMod2 repository to pull the latest changes to the code.

Best, Umair