summaryrefslogtreecommitdiff
path: root/training
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-08 08:42:11 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-08 08:42:11 +0200
commit57d785164757b812a9270b88f76ea88861472b1f (patch)
tree944b3073f8b994dac4002263817b6345629868f0 /training
parent80de5c023cb7740022a9b13b5f09b71d9e85f8e7 (diff)
Add ctc loss config
Diffstat (limited to 'training')
-rw-r--r--training/conf/criterion/ctc.yaml2
1 files changed, 2 insertions, 0 deletions
diff --git a/training/conf/criterion/ctc.yaml b/training/conf/criterion/ctc.yaml
new file mode 100644
index 0000000..0677c06
--- /dev/null
+++ b/training/conf/criterion/ctc.yaml
@@ -0,0 +1,2 @@
+_target_: torch.nn.CTCLoss
+zero_infinity: true