diff --git a/chapters/en/unit2/cnns/googlenet.mdx b/chapters/en/unit2/cnns/googlenet.mdx index 4e3ae0d36..8e8aa163f 100644 --- a/chapters/en/unit2/cnns/googlenet.mdx +++ b/chapters/en/unit2/cnns/googlenet.mdx @@ -95,7 +95,10 @@ class InceptionModule(nn.Module): def __init__(self, in_channels, n1x1, n3x3red, n3x3, n5x5red, n5x5, pool_proj): super(InceptionModule, self).__init__() - self.b1 = BaseConv2d(in_channels, n1x1, kernel_size=1) + self.b1 = nn.Sequential( + nn.Conv2d(in_channels, n1x1, kernel_size=1), + nn.ReLU(True), + ) self.b2 = nn.Sequential( BaseConv2d(in_channels, n3x3red, kernel_size=1), @@ -228,4 +231,4 @@ class GoogLeNet(nn.Module): return x, aux1, aux2 else: return x -``` \ No newline at end of file +``` diff --git a/requirements.txt b/requirements.txt index 7da976f5f..5c6a1d317 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,9 +1,9 @@ -transformers==4.34.0 -timm==0.9.7 -evaluate==0.4.1 -datasets==2.14.5 +#transformers==4.34.0 +#timm==0.9.7 +#valuate==0.4.1 +#datasets==2.14.5 jupyter==1.0.0 ipywidgets==8.1.1 -torch==2.1.0 +#torch==2.1.0 albumentations==1.3.1 matplotlib==3.7.1 \ No newline at end of file diff --git a/utils/code_formatter.py b/utils/code_formatter.py index 4525fcf06..3dc7d9a30 100644 --- a/utils/code_formatter.py +++ b/utils/code_formatter.py @@ -93,4 +93,4 @@ def format_all_files(check_only=False): ) args = parser.parse_args() - format_all_files(check_only=args.check_only) + format_all_files(check_only=args.check_only) \ No newline at end of file