hi，i am very much appreciate your work. I want to know why Comment “momentum_update_key_encoder” in function “forward” during the forward pass. The code is line 224 in model.py

# Perform momentum update during the forward pass

momentum_update_key_encoder(self.vision_base_q, self.vision_base_k, self.vision_head_intra_q,

self.vision_head_intra_k, self.m)

momentum_update_key_encoder(self.tactile_base_q, self.tactile_base_k, self.tactile_head_intra_q,

self.tactile_head_intra_k, self.m)

@ligerfotis

Hello, pre-trained weight can open source. thanks

After downloading the dataset Calandra's Dataset, the data could not be prepared by the generate_dataset.py script, which could not satisfy the data preparation requirements for the training