checkpoint_load

Не работает загрузка чекпоинтов

Traceback (most recent call last):
  File "/home/pjdanilov/.conda/envs/pdanilov/bin/catalyst-dl", line 8, in <module>
    sys.exit(main())
  File "/home/pjdanilov/.conda/envs/pdanilov/lib/python3.8/site-packages/catalyst/dl/__main__.py", line 41, in main
    COMMANDS[args.command].main(args, uargs)
  File "/home/pjdanilov/.conda/envs/pdanilov/lib/python3.8/site-packages/catalyst/dl/scripts/run.py", line 120, in main
    distributed_run(args.distributed, main_worker, args, unknown_args)
  File "/home/pjdanilov/.conda/envs/pdanilov/lib/python3.8/site-packages/catalyst/utils/distributed.py", line 132, in distributed_run
    worker_fn(*args, **kwargs)
  File "/home/pjdanilov/.conda/envs/pdanilov/lib/python3.8/site-packages/catalyst/dl/scripts/run.py", line 115, in main_worker
    runner.run_experiment(experiment, check=check_run)
  File "/home/pjdanilov/.conda/envs/pdanilov/lib/python3.8/site-packages/catalyst/contrib/dl/runner/wandb.py", line 142, in run_experiment
    super().run_experiment(experiment=experiment, check=check)
  File "/home/pjdanilov/.conda/envs/pdanilov/lib/python3.8/site-packages/catalyst/core/runner.py", line 389, in run_experiment
    self._run_event("exception", moment=None)
  File "/home/pjdanilov/.conda/envs/pdanilov/lib/python3.8/site-packages/catalyst/core/runner.py", line 234, in _run_event
    getattr(logger, fn_name)(self.state)
  File "/home/pjdanilov/.conda/envs/pdanilov/lib/python3.8/site-packages/catalyst/dl/callbacks/misc.py", line 152, in on_exception
    raise exception
  File "/home/pjdanilov/.conda/envs/pdanilov/lib/python3.8/site-packages/catalyst/core/runner.py", line 381, in run_experiment
    self._run_stage(stage)
  File "/home/pjdanilov/.conda/envs/pdanilov/lib/python3.8/site-packages/catalyst/core/runner.py", line 342, in _run_stage
    self._run_event("stage", moment="start")
  File "/home/pjdanilov/.conda/envs/pdanilov/lib/python3.8/site-packages/catalyst/core/runner.py", line 228, in _run_event
    getattr(callback, fn_name)(self.state)
  File "/home/pjdanilov/.conda/envs/pdanilov/lib/python3.8/site-packages/catalyst/core/callbacks/checkpoint.py", line 222, in on_stage_start
    state.resume = None
  File "/home/pjdanilov/.conda/envs/pdanilov/lib/python3.8/site-packages/catalyst/utils/tools/frozen_class.py", line 12, in __setattr__
    raise TypeError("%r is a frozen class" % self)
TypeError: <catalyst.dl.core.state.State object at 0x7f32d52a4f40> is a frozen class