From 9353a39a18d0542afc177cd134f33f0756820a7d Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Thu, 9 Jun 2022 22:33:34 +0200
Subject: Remove abstract lightning module

---
 text_recognizer/networks/cnn.py | 26 --------------------------
 1 file changed, 26 deletions(-)
 delete mode 100644 text_recognizer/networks/cnn.py

(limited to 'text_recognizer/networks/cnn.py')

diff --git a/text_recognizer/networks/cnn.py b/text_recognizer/networks/cnn.py
deleted file mode 100644
index 5e2a7f4..0000000
--- a/text_recognizer/networks/cnn.py
+++ /dev/null
@@ -1,26 +0,0 @@
-"""Simple convolutional network."""
-import torch
-from torch import nn, Tensor
-
-
-class CNN(nn.Module):
-    def __init__(self, channels: int, depth: int) -> None:
-        super().__init__()
-        self.layers = self._build(channels, depth)
-
-    def _build(self, channels: int, depth: int) -> nn.Sequential:
-        layers = []
-        for i in range(depth):
-            layers.append(
-                nn.Conv2d(
-                    in_channels=1 if i == 0 else channels,
-                    out_channels=channels,
-                    kernel_size=3,
-                    stride=2,
-                )
-            )
-            layers.append(nn.Mish(inplace=True))
-        return nn.Sequential(*layers)
-
-    def forward(self, x: Tensor) -> Tensor:
-        return self.layers(x)
-- 
cgit v1.2.3-70-g09d2