Upload 2 files

model_config.yaml

@@ -0,0 +1,165 @@
+sample_rate: 16000
+timesteps: 64
+repeat: 2
+dropout: 0.0
+kernel_size_factor: 1.0
+labels:
+- background
+- speech
+train_ds:
+  vad_stream: false
+  manifest_filepath: /home/fjia/code/manifest64/train.json
+  sample_rate: 16000
+  labels:
+  - background
+  - speech
+  batch_size: 128
+  num_workers: 20
+  shuffle: true
+  augmentor:
+    shift:
+      prob: 0.8
+      min_shift_ms: -5.0
+      max_shift_ms: 5.0
+    white_noise:
+      prob: 0.8
+      min_level: -90
+      max_level: -46
+validation_ds:
+  vad_stream: false
+  manifest_filepath: /home/fjia/code/manifest64/validation.json
+  sample_rate: 16000
+  labels:
+  - background
+  - speech
+  batch_size: 128
+  shuffle: false
+  val_loss_idx: 0
+  num_workers: 20
+test_ds:
+  vad_stream: false
+  manifest_filepath: null
+  sample_rate: 16000
+  labels:
+  - background
+  - speech
+  batch_size: 128
+  shuffle: false
+  test_loss_idx: 0
+  num_workers: 20
+preprocessor:
+  cls: nemo.collections.asr.modules.AudioToMFCCPreprocessor
+  params:
+    window_size: 0.025
+    window_stride: 0.01
+    window: hann
+    n_mels: 64
+    n_mfcc: 64
+    n_fft: 512
+spec_augment:
+  cls: nemo.collections.asr.modules.SpectrogramAugmentation
+  params:
+    freq_masks: 2
+    time_masks: 2
+    freq_width: 15
+    time_width: 25
+    rect_masks: 5
+    rect_time: 25
+    rect_freq: 15
+encoder:
+  cls: nemo.collections.asr.modules.ConvASREncoder
+  params:
+    feat_in: 64
+    activation: relu
+    conv_mask: true
+    jasper:
+    - filters: 128
+      repeat: 1
+      kernel:
+      - 11
+      stride:
+      - 1
+      dilation:
+      - 1
+      dropout: 0.0
+      residual: false
+      separable: true
+      kernel_size_factor: 1.0
+    - filters: 64
+      repeat: 2
+      kernel:
+      - 13
+      stride:
+      - 1
+      dilation:
+      - 1
+      dropout: 0.0
+      residual: true
+      separable: true
+      kernel_size_factor: 1.0
+    - filters: 64
+      repeat: 2
+      kernel:
+      - 15
+      stride:
+      - 1
+      dilation:
+      - 1
+      dropout: 0.0
+      residual: true
+      separable: true
+      kernel_size_factor: 1.0
+    - filters: 64
+      repeat: 2
+      kernel:
+      - 17
+      stride:
+      - 1
+      dilation:
+      - 1
+      dropout: 0.0
+      residual: true
+      separable: true
+      kernel_size_factor: 1.0
+    - filters: 128
+      repeat: 1
+      kernel:
+      - 29
+      stride:
+      - 1
+      dilation:
+      - 2
+      dropout: 0.0
+      residual: false
+      separable: true
+      kernel_size_factor: 1.0
+    - filters: 128
+      repeat: 1
+      kernel:
+      - 1
+      stride:
+      - 1
+      dilation:
+      - 1
+      dropout: 0.0
+      residual: false
+decoder:
+  cls: nemo.collections.asr.modules.ConvASRDecoderClassification
+  params:
+    feat_in: 128
+    num_classes: 2
+    return_logits: true
+    pooling_type: avg
+optim:
+  name: sgd
+  lr: 0.01
+  weight_decay: 0.001
+  sched:
+    name: PolynomialHoldDecayAnnealing
+    power: 2.0
+    warmup_ratio: 0.05
+    hold_ratio: 0.45
+    min_lr: 0.001
+    last_epoch: -1
+  momentum: 0.9
+target: nemo.collections.asr.models.classification_models.EncDecClassificationModel

model_weights.ckpt

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