Error while running SAC Agent

Hello Folks,

Getting the following error while running SAC agent on local.

[Controller] Starting simulator
sudo: unknown user: ubuntu
sudo: unable to initialize policy plugin
[Controller] Setting level to Thruxton
[Controller] Setting to RL mode for vehicle: 0
[Controller] Enabling CameraFrontRGB
[Controller] Enabling CameraFrontRGB
[Controller] Setting: ArrivalVehicleDriver parameters for vehicle: 0
[Controller] Setting: CameraFrontRGB parameters for vehicle: 0
2022-01-11 23:20:55,292 [INFO] Using random seed: 0
Traceback (most recent call last):
  File "rollout.py", line 44, in <module>
    run_evaluation()
  File "rollout.py", line 35, in run_evaluation
    evaluator.init_agent()
  File "/home/allthatido/L2R Challenge/l2r-starter-kit/evaluator/evaluator.py", line 56, in init_agent
    self.agent = self.submission_config.agent()
  File "/home/allthatido/L2R Challenge/l2r-starter-kit/agents/sac_agent.py", line 49, in __init__
    self.setup_vision_encoder()
  File "/home/allthatido/L2R Challenge/l2r-starter-kit/agents/sac_agent.py", line 115, in setup_vision_encoder
    torch.load(self.cfg["vae"]["vae_chkpt_statedict"], map_location=DEVICE)
  File "/home/allthatido/.local/lib/python3.8/site-packages/torch/serialization.py", line 608, in load
    return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
  File "/home/allthatido/.local/lib/python3.8/site-packages/torch/serialization.py", line 777, in _legacy_load
    magic_number = pickle_module.load(f, **pickle_load_args)
_pickle.UnpicklingError: invalid load key, 'v'.

Any help is appreciated.

1 Like

Maybe it got corrupted during download. You can try redownloading it manually:

2 Likes

Thanks. That seems to have worked. Can you also please explain your intuition behind reaching that conclusion ?

1 Like

I just googled it. But I guessed before that it was most likely either your system not being compatible with the saved file or a download corruption.

1 Like