diff --git a/src/rl_sar/models/actor.pt b/src/rl_sar/models/actor.pt
index 719cf73..912342c 100644
Binary files a/src/rl_sar/models/actor.pt and b/src/rl_sar/models/actor.pt differ
diff --git a/src/rl_sar/models/encoder.pt b/src/rl_sar/models/encoder.pt
index 74f8d12..3a8edb6 100644
Binary files a/src/rl_sar/models/encoder.pt and b/src/rl_sar/models/encoder.pt differ
diff --git a/src/rl_sar/models/vq_layer.pt b/src/rl_sar/models/vq_layer.pt
index ba6d318..21bcc92 100644
Binary files a/src/rl_sar/models/vq_layer.pt and b/src/rl_sar/models/vq_layer.pt differ