renaming model.py to models.py

Niccolo-Ajroldi · Feb 10, 2023 · 80a7849 · 80a7849
1 parent fc02196
commit 80a7849
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 6 deletions.
diff --git a/...ch_conformer/librispeech_pytorch/model.py → ...h_conformer/librispeech_pytorch/models.py b/...ch_conformer/librispeech_pytorch/model.py → ...h_conformer/librispeech_pytorch/models.py
diff --git a/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_pytorch/workload.py b/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_pytorch/workload.py
@@ -18,7 +18,7 @@
 from algorithmic_efficiency.workloads.librispeech_conformer.input_pipeline import \
     LibriSpeechDataset
 from algorithmic_efficiency.workloads.librispeech_conformer.librispeech_pytorch import \
-    model as conformer_model
+    models as conformer_model
 
 USE_PYTORCH_DDP, RANK, DEVICE, N_GPUS = pytorch_utils.pytorch_setup()
 

diff --git a/...h_deepspeech/librispeech_pytorch/model.py → ..._deepspeech/librispeech_pytorch/models.py b/...h_deepspeech/librispeech_pytorch/model.py → ..._deepspeech/librispeech_pytorch/models.py
diff --git a/algorithmic_efficiency/workloads/librispeech_deepspeech/librispeech_pytorch/workload.py b/algorithmic_efficiency/workloads/librispeech_deepspeech/librispeech_pytorch/workload.py
@@ -6,13 +6,13 @@
 from algorithmic_efficiency import param_utils
 from algorithmic_efficiency import spec
 from algorithmic_efficiency.pytorch_utils import pytorch_setup
-from algorithmic_efficiency.workloads.librispeech_conformer.librispeech_pytorch.model import \
+from algorithmic_efficiency.workloads.librispeech_conformer.librispeech_pytorch.models import \
     initialize
 from algorithmic_efficiency.workloads.librispeech_conformer.librispeech_pytorch.workload import \
     LibriSpeechConformerWorkload
-from algorithmic_efficiency.workloads.librispeech_deepspeech.librispeech_pytorch.model import \
+from algorithmic_efficiency.workloads.librispeech_deepspeech.librispeech_pytorch.models import \
     DeepspeechConfig
-from algorithmic_efficiency.workloads.librispeech_deepspeech.librispeech_pytorch.model import \
+from algorithmic_efficiency.workloads.librispeech_deepspeech.librispeech_pytorch.models import \
     DeepspeechEncoderDecoder
 
 USE_PYTORCH_DDP, RANK, DEVICE, N_GPUS = pytorch_setup()

diff --git a/tests/test_num_params.py b/tests/test_num_params.py
@@ -25,9 +25,9 @@
     Conformer as JaxConformer
 from algorithmic_efficiency.workloads.librispeech_conformer.librispeech_jax.models import \
     ConformerConfig as JaxConformerConfig
-from algorithmic_efficiency.workloads.librispeech_conformer.librispeech_pytorch.model import \
+from algorithmic_efficiency.workloads.librispeech_conformer.librispeech_pytorch.models import \
     ConformerConfig as PytorchConformerConfig
-from algorithmic_efficiency.workloads.librispeech_conformer.librispeech_pytorch.model import \
+from algorithmic_efficiency.workloads.librispeech_conformer.librispeech_pytorch.models import \
     ConformerEncoderDecoder as PytorchConformer
 from algorithmic_efficiency.workloads.mnist.mnist_jax.workload import \
     _Model as JaxMLP