04_hil_serl.py run error

[wadefrank]

python 04_hil_serl.py
Traceback (most recent call last):
  File "/home/wade/Documents/SecondBrain/Codes/Robot_Learning_A_Tutorial/ch3_ReinforcementLearning/04_hil_serl.py", line 39, in <module>
    reward_classifier = Classifier.from_pretrained(reward_classifier_id)
  File "/home/wade/wade/Code/robotics/lerobot/src/lerobot/policies/pretrained.py", line 125, in from_pretrained
    policy = cls._load_as_safetensor(instance, model_file, config.device, strict=False)
  File "/home/wade/wade/Code/robotics/lerobot/src/lerobot/policies/pretrained.py", line 145, in _load_as_safetensor
    missing_keys, unexpected_keys = load_model_as_safetensor(model, model_file, **kwargs)
  File "/home/wade/anaconda3/envs/lerobot_v0.4.0/lib/python3.10/site-packages/safetensors/torch.py", line 285, in load_model
    assert torch_unexpected == unexpected, f"{torch_unexpected} != {unexpected}"
AssertionError: {'encoders.observation_images_top.0.encoder.stages.1.layers.0.layer.1.normalization.num_batches_tracked', 'encoders.observation_images_top.0.encoder.stages.0.layers.0.layer.0.normalization.num_batches_tracked', 'encoders.observation_images_top.0.encoder.stages.3.layers.0.shortcut.normalization.num_batches_tracked', 'encoders.observation_images_top.0.encoder.stages.1.layers.0.shortcut.normalization.num_batches_tracked', 'encoders.observation_images_top.0.encoder.stages.2.layers.1.layer.1.normalization.num_batches_tracked', 'encoders.observation_images_top.0.encoder.stages.3.layers.1.layer.1.normalization.num_batches_tracked', 'encoders.observation_images_top.0.encoder.stages.3.layers.0.layer.0.normalization.num_batches_tracked', 'encoders.observation_images_top.0.embedder.embedder.normalization.num_batches_tracked', 'encoders.observation_images_top.0.encoder.stages.1.layers.0.layer.0.normalization.num_batches_tracked', 'encoders.observation_images_top.0.encoder.stages.3.layers.0.layer.1.normalization.num_batches_tracked', 'encoders.observation_images_top.0.encoder.stages.3.layers.1.layer.0.normalization.num_batches_tracked', 'encoders.observation_images_top.0.encoder.stages.2.layers.0.layer.0.normalization.num_batches_tracked', 'encoders.observation_images_top.0.encoder.stages.0.layers.1.layer.0.normalization.num_batches_tracked', 'encoders.observation_images_top.0.encoder.stages.1.layers.1.layer.0.normalization.num_batches_tracked', 'encoders.observation_images_top.0.encoder.stages.0.layers.1.layer.1.normalization.num_batches_tracked', 'encoders.observation_images_top.0.encoder.stages.2.layers.1.layer.0.normalization.num_batches_tracked', 'encoders.observation_images_top.0.encoder.stages.1.layers.1.layer.1.normalization.num_batches_tracked', 'encoders.observation_images_top.0.encoder.stages.2.layers.0.shortcut.normalization.num_batches_tracked', 'encoders.observation_images_top.0.encoder.stages.2.layers.0.layer.1.normalization.num_batches_tracked', 'encoders.observation_images_top.0.encoder.stages.0.layers.0.layer.1.normalization.num_batches_tracked'} != set()

how to fix?