Upload model

config.json

@@ -7,7 +7,7 @@
   "model": {
     "cls": {
       "combined_dim": 128,
-      "combined_input": 384,
+      "combined_input": 256,
       "n_layers": 4,
       "num_classes": 1,
       "skip_in": [
@@ -16,7 +16,7 @@
       "weight_norm": true
     },
     "freq": {
-      "n_layers": 4,
+      "n_layers": 2,
       "skip_in": [
         80
       ],
@@ -28,7 +28,7 @@
     "pos": {
       "hidden_dim": 256,
       "input_dim": 300,
-      "n_layers": 4,
+      "n_layers": 2,
       "output_dim": 128,
       "skip_in": [
         80
@@ -40,7 +40,7 @@
       "hidden_dim": 256,
       "input_dim": 768,
       "lstm_in": 768,
-      "n_layers": 4,
+      "n_layers": 2,
       "output_dim": 128,
       "skip_in": [
         80
@@ -53,7 +53,7 @@
   "torch_dtype": "float32",
   "train": {
     "anneal_end": 5,
-    "batch_size": 64,
+    "batch_size": 32,
     "end_iter": 10,
     "learning_rate": 0.0005,
     "learning_rate_alpha": 0.05,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:741418e39e9eecde888539dfe7d95bb19c28e9ca7d3b23ffacc996fcb930252c
-size 480558480
+oid sha256:11b7c11ebe62383c41a5977e808bd1820ad9d9aa167d6f42c3d15ef3f2f6f6e6
+size 477329272