Update epochs in train_vq_transformer and add VQ Transformer model files

train_vq_transformer.py

@@ -38,7 +38,7 @@ def train_vq_transformer():
         max_position_embeddings=50,  # 1 label + 49 tokens
         label_offset=512,  # Labels are tokens 512-521
         batch_size=32,
-        epochs=5,
+        epochs=10, #5
         lr=1e-3,
         warmup_steps=500,
         device=device

vq_transformer_model/config.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:039711d65d809e22b49af2a55d4f0a6786c29d3a90916577002c5fcef4cb91d6
-size 1116
+oid sha256:d32ec3ecf4ce9ed2fdab28e25eab82913de96c965b281ba5cf148337ea90a594
+size 1511

vq_transformer_model/model.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1c8818673ab2c30c139a3f51ca6516beb74c32b93b9e2d43b6ec65b373971f38
-size 1135532
+oid sha256:7e84c4c3bddae645a00668fb3d51fcc68554aa83c9810e930f392c47bd0eccd6
+size 1135925