update

rwth-i6 · Oct 20, 2023 · f26d94c · f26d94c
1 parent 370fb10
commit f26d94c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 9 deletions.
diff --git a/i6_models/parts/frontend/generic_frontend.py b/i6_models/parts/frontend/generic_frontend.py
@@ -31,7 +31,8 @@ class GenericFrontendV1Config(ModelConfiguration):
         in_features: number of input features to module
         layer_ordering: the ordering of the front end layer sequences, the ordering element must be selected from FrontendLayerType
             e.g. the ordering of VGG4LayerActFrontendV1 would be [FrontendLayerType.Conv2d, FrontendLayerType.Activation,
-            FrontendLayerType.Pool2d, FrontendLayerType.Conv2d, FrontendLayerType.Conv2d, FrontendLayerType.Activation, FrontendLayerType.Pool2d]
+            FrontendLayerType.Pool2d, FrontendLayerType.Conv2d, FrontendLayerType.Conv2d, FrontendLayerType.Activation,
+            FrontendLayerType.Pool2d]
         conv_kernel_sizes: kernel sizes for each conv layer
         conv_strides: stride sizes for each conv layer
         conv_paddings: paddings sizes for each conv layer
@@ -102,7 +103,8 @@ class GenericFrontendV1(nn.Module):
     def __init__(self, model_cfg: GenericFrontendV1Config):
         """
         Generic Front-End
-        can be used to generate customized frontend by combine convolutional and pooling layers, as well as activation functions different
+        can be used to generate customized frontend by combining convolutional and pooling layers, as well as activation
+        functions differently
 
         To get the ESPnet case, for example Conv2dSubsampling6, use these options
             layer_ordering = [FrontendLayerType.Conv2d, FrontendLayerType.Conv2d]

diff --git a/tests/test_generic_frontend.py b/tests/test_generic_frontend.py
@@ -6,10 +6,6 @@
 import torch
 from torch import nn
 
-import sys
-
-sys.path.insert(0, "/Users/jxu/Desktop/PR/i6_models")
-
 from i6_models.parts.frontend.generic_frontend import FrontendLayerType, GenericFrontendV1, GenericFrontendV1Config
 
 
@@ -169,6 +165,3 @@ def get_output_shape(test_parameters: GenericFrontendV1TestParams):
         )
 
     return
-
-
-test_generic_frontend_v1()