v2

Signed-off-by: Diwank Singh <[email protected]>

2_Asym/140225464223632_Dense/config.json

@@ -0,0 +1 @@
+{"in_features": 1024, "out_features": 2048, "bias": true, "activation_function": "torch.nn.modules.activation.Tanh"}

2_Asym/140225464223632_Dense/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e09d27f6e37e79b5889f99320b29eab442407414661f86706af8946f783d0239
+size 8398460

2_Asym/140225464231312_Dense/config.json

@@ -0,0 +1 @@
+{"in_features": 2048, "out_features": 2048, "bias": true, "activation_function": "torch.nn.modules.activation.Tanh"}

2_Asym/140225464231312_Dense/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:75db6e76d6b5838f1cd255ba48a1fffb13b807985583df180effbfb73abbfc92
+size 16787068

2_Asym/140225464232416_Dense/config.json

@@ -0,0 +1 @@
+{"in_features": 2048, "out_features": 1024, "bias": true, "activation_function": "torch.nn.modules.activation.Tanh"}

2_Asym/140225464232416_Dense/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f73d9d5ee30e1c9c367620f4b32b59133841a9aa66c34c213c4eceeed1d91739
+size 8394364

2_Asym/140225464233232_Dropout/config.json

@@ -0,0 +1 @@
+{"dropout": 0.1}

2_Asym/140225464235392_Dense/config.json

@@ -0,0 +1 @@
+{"in_features": 2048, "out_features": 2048, "bias": true, "activation_function": "torch.nn.modules.activation.Tanh"}

2_Asym/140225464235392_Dense/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:53306f081b130f877d2d20e684b403951a08cf410c6854f2fc23f49427d5ba85
+size 16787068

2_Asym/140225464235968_Dense/config.json

@@ -0,0 +1 @@
+{"in_features": 2048, "out_features": 1024, "bias": true, "activation_function": "torch.nn.modules.activation.Tanh"}

2_Asym/140225464235968_Dense/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:762e434b5b7dcfef9ae42841a4193578ecb74d750c5efa5433fd0b129883511f
+size 8394364

2_Asym/140225464236064_Dense/config.json

@@ -0,0 +1 @@
+{"in_features": 2048, "out_features": 2048, "bias": true, "activation_function": "torch.nn.modules.activation.Tanh"}

2_Asym/140225464236064_Dense/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b7cc7b864e5a7d41a179fa8660606d0d65e7a0ae3a3ac2076b810ee07234861c
+size 16787068

2_Asym/140225464236784_Dense/config.json

@@ -0,0 +1 @@
+{"in_features": 1024, "out_features": 2048, "bias": true, "activation_function": "torch.nn.modules.activation.Tanh"}

2_Asym/140225464236784_Dense/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f2759d8139196720ab46d120d3225231d160bd9fa74f39d7f59bab8a447155db
+size 8398460

2_Asym/140225507851568_Dropout/config.json

@@ -0,0 +1 @@
+{"dropout": 0.1}

2_Asym/140228101622896_Dense/config.json

@@ -0,0 +1 @@
+{"in_features": 2048, "out_features": 2048, "bias": true, "activation_function": "torch.nn.modules.activation.Tanh"}

2_Asym/140228101622896_Dense/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5c82ba0624007b1e5b7f281fdafa92e7106f7c7f954de0617fcf3ff261bd154f
+size 16787068

2_Asym/config.json

@@ -1,26 +1,34 @@
 {
   "types": {
-    "139759749873328_Dense": "sentence_transformers.models.Dense",
-    "139771065711008_Dense": "sentence_transformers.models.Dense",
-    "139762386917776_Dense": "sentence_transformers.models.Dense",
-    "139762386917200_Normalize": "sentence_transformers.models.Normalize",
-    "139771065971808_Dense": "sentence_transformers.models.Dense",
-    "139759749870880_Dense": "sentence_transformers.models.Dense",
-    "139759749869296_Dense": "sentence_transformers.models.Dense",
-    "139759749864928_Normalize": "sentence_transformers.models.Normalize"
+    "140225464236784_Dense": "sentence_transformers.models.Dense",
+    "140228101622896_Dense": "sentence_transformers.models.Dense",
+    "140225507851568_Dropout": "sentence_transformers.models.Dropout",
+    "140225464235392_Dense": "sentence_transformers.models.Dense",
+    "140225464235968_Dense": "sentence_transformers.models.Dense",
+    "140225464231168_Normalize": "sentence_transformers.models.Normalize",
+    "140225464223632_Dense": "sentence_transformers.models.Dense",
+    "140225464236064_Dense": "sentence_transformers.models.Dense",
+    "140225464233232_Dropout": "sentence_transformers.models.Dropout",
+    "140225464231312_Dense": "sentence_transformers.models.Dense",
+    "140225464232416_Dense": "sentence_transformers.models.Dense",
+    "140225464235536_Normalize": "sentence_transformers.models.Normalize"
   },
   "structure": {
     "dialog": [
-      "139759749873328_Dense",
-      "139771065711008_Dense",
-      "139762386917776_Dense",
-      "139762386917200_Normalize"
+      "140225464236784_Dense",
+      "140228101622896_Dense",
+      "140225507851568_Dropout",
+      "140225464235392_Dense",
+      "140225464235968_Dense",
+      "140225464231168_Normalize"
     ],
     "fact": [
-      "139771065971808_Dense",
-      "139759749870880_Dense",
-      "139759749869296_Dense",
-      "139759749864928_Normalize"
+      "140225464223632_Dense",
+      "140225464236064_Dense",
+      "140225464233232_Dropout",
+      "140225464231312_Dense",
+      "140225464232416_Dense",
+      "140225464235536_Normalize"
     ]
   },
   "parameters": {

README.md

@@ -53,12 +53,12 @@ The model was trained with the parameters:
 
 **Loss**:
 
-`sentence_transformers.losses.OnlineContrastiveLoss.OnlineContrastiveLoss` 
+`sentence_transformers.losses.CosineSimilarityLoss.CosineSimilarityLoss` 
 
 Parameters of the fit()-Method:
 ```
 {
-    "epochs": 6,
+    "epochs": 4,
     "evaluation_steps": 2000,
     "evaluator": "sentence_transformers.evaluation.EmbeddingSimilarityEvaluator.EmbeddingSimilarityEvaluator",
     "max_grad_norm": 1,
@@ -83,12 +83,20 @@ SentenceTransformer(
   (2): Asym(
     (dialog-0): Dense({'in_features': 1024, 'out_features': 2048, 'bias': True, 'activation_function': 'torch.nn.modules.activation.Tanh'})
     (dialog-1): Dense({'in_features': 2048, 'out_features': 2048, 'bias': True, 'activation_function': 'torch.nn.modules.activation.Tanh'})
-    (dialog-2): Dense({'in_features': 2048, 'out_features': 1024, 'bias': True, 'activation_function': 'torch.nn.modules.activation.Tanh'})
-    (dialog-3): Normalize()
+    (dialog-2): Dropout(
+      (dropout_layer): Dropout(p=0.1, inplace=False)
+    )
+    (dialog-3): Dense({'in_features': 2048, 'out_features': 2048, 'bias': True, 'activation_function': 'torch.nn.modules.activation.Tanh'})
+    (dialog-4): Dense({'in_features': 2048, 'out_features': 1024, 'bias': True, 'activation_function': 'torch.nn.modules.activation.Tanh'})
+    (dialog-5): Normalize()
     (fact-0): Dense({'in_features': 1024, 'out_features': 2048, 'bias': True, 'activation_function': 'torch.nn.modules.activation.Tanh'})
     (fact-1): Dense({'in_features': 2048, 'out_features': 2048, 'bias': True, 'activation_function': 'torch.nn.modules.activation.Tanh'})
-    (fact-2): Dense({'in_features': 2048, 'out_features': 1024, 'bias': True, 'activation_function': 'torch.nn.modules.activation.Tanh'})
-    (fact-3): Normalize()
+    (fact-2): Dropout(
+      (dropout_layer): Dropout(p=0.1, inplace=False)
+    )
+    (fact-3): Dense({'in_features': 2048, 'out_features': 2048, 'bias': True, 'activation_function': 'torch.nn.modules.activation.Tanh'})
+    (fact-4): Dense({'in_features': 2048, 'out_features': 1024, 'bias': True, 'activation_function': 'torch.nn.modules.activation.Tanh'})
+    (fact-5): Normalize()
   )
 )
 ```