From b44de0e11281c723ec426f8bec8ca0897ecfe3ff Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Sun, 21 Nov 2021 21:34:53 +0100
Subject: Remove VQVAE stuff, did not work...

---
 training/conf/criterion/mae.yaml        |  2 --
 training/conf/criterion/mse.yaml        |  2 --
 training/conf/criterion/vqgan_loss.yaml | 11 -----------
 3 files changed, 15 deletions(-)
 delete mode 100644 training/conf/criterion/mae.yaml
 delete mode 100644 training/conf/criterion/mse.yaml
 delete mode 100644 training/conf/criterion/vqgan_loss.yaml

(limited to 'training/conf/criterion')

diff --git a/training/conf/criterion/mae.yaml b/training/conf/criterion/mae.yaml
deleted file mode 100644
index cb07467..0000000
--- a/training/conf/criterion/mae.yaml
+++ /dev/null
@@ -1,2 +0,0 @@
-_target_: torch.nn.L1Loss
-reduction: mean
diff --git a/training/conf/criterion/mse.yaml b/training/conf/criterion/mse.yaml
deleted file mode 100644
index ffd1403..0000000
--- a/training/conf/criterion/mse.yaml
+++ /dev/null
@@ -1,2 +0,0 @@
-_target_: torch.nn.MSELoss
-reduction: mean
diff --git a/training/conf/criterion/vqgan_loss.yaml b/training/conf/criterion/vqgan_loss.yaml
deleted file mode 100644
index 34a67ae..0000000
--- a/training/conf/criterion/vqgan_loss.yaml
+++ /dev/null
@@ -1,11 +0,0 @@
-_target_: text_recognizer.criterion.vqgan_loss.VQGANLoss
-reconstruction_loss:
-  _target_: torch.nn.MSELoss
-  reduction: mean
-discriminator:
-  _target_: text_recognizer.criterions.n_layer_discriminator.NLayerDiscriminator
-  in_channels: 1
-  num_channels: 32
-  num_layers: 3
-vq_loss_weight: 1.0
-discriminator_weight: 1.0
-- 
cgit v1.2.3-70-g09d2