comfi-fast-grnn-torch 0.0.2__py3-none-any.whl → 0.0.3__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- comfi_fast_grnn_torch/__init__.py +1 -0
- {comfi_fast_grnn_torch-0.0.2.dist-info → comfi_fast_grnn_torch-0.0.3.dist-info}/METADATA +3 -15
- comfi_fast_grnn_torch-0.0.3.dist-info/RECORD +6 -0
- comfi_fast_grnn_torch-0.0.2.dist-info/RECORD +0 -6
- {comfi_fast_grnn_torch-0.0.2.dist-info → comfi_fast_grnn_torch-0.0.3.dist-info}/WHEEL +0 -0
- {comfi_fast_grnn_torch-0.0.2.dist-info → comfi_fast_grnn_torch-0.0.3.dist-info}/top_level.txt +0 -0
|
@@ -0,0 +1 @@
|
|
|
1
|
+
from .ComfiFastGRNN import ComfiFastGRNN
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.4
|
|
2
2
|
Name: comfi_fast_grnn_torch
|
|
3
|
-
Version: 0.0.
|
|
3
|
+
Version: 0.0.3
|
|
4
4
|
Summary: A PyTorch implementation of Comfi-FastGRNN
|
|
5
5
|
Author-email: Nicolas Arrieta Larraza <NIAL@bang-olufsen.dk>, Niels de Koeijer <NEMK@bang-olufsen.dk>
|
|
6
6
|
License: MIT
|
|
@@ -12,7 +12,7 @@ Requires-Dist: libsegmenter==1.0.4
|
|
|
12
12
|
Requires-Dist: torchinfo==1.8.0
|
|
13
13
|
Requires-Dist: CRM_pytorch==0.1.0
|
|
14
14
|
|
|
15
|
-
# fast-
|
|
15
|
+
# comfi-fast-grnn-torch
|
|
16
16
|
Implements Comfi-FastGRNN in torch.
|
|
17
17
|
|
|
18
18
|
## Usage
|
|
@@ -26,21 +26,9 @@ Here is how to use the layer with default settings in a standard training loop:
|
|
|
26
26
|
import torch
|
|
27
27
|
from comfi_fast_grnn_torch import ComfiFastGRNN
|
|
28
28
|
|
|
29
|
-
|
|
30
|
-
# batch_first=True is the default for this implementation
|
|
31
|
-
model = ComfiFastGRNN(
|
|
29
|
+
comfi_fgrnn = ComfiFastGRNN(
|
|
32
30
|
input_size=32,
|
|
33
31
|
hidden_size=64,
|
|
34
32
|
num_layers=1
|
|
35
33
|
)
|
|
36
|
-
|
|
37
|
-
# 2. Create dummy input: (Batch Size, Sequence Length, Input Size)
|
|
38
|
-
x = torch.randn(10, 50, 32)
|
|
39
|
-
|
|
40
|
-
# 3. Forward pass
|
|
41
|
-
# Returns output (all timesteps) and final hidden state
|
|
42
|
-
output, h_n = model(x)
|
|
43
|
-
|
|
44
|
-
print(f"Output shape: {output.shape}") # torch.Size([10, 50, 64])
|
|
45
|
-
print(f"Hidden state shape: {h_n.shape}") # torch.Size([1, 10, 64])
|
|
46
34
|
```
|
|
@@ -0,0 +1,6 @@
|
|
|
1
|
+
comfi_fast_grnn_torch/ComfiFastGRNN.py,sha256=uUX5h1ilLcicWtJt1BSPVGZmUOaeltWpjC6e4wq4pIE,10085
|
|
2
|
+
comfi_fast_grnn_torch/__init__.py,sha256=d7Ar1_hIgF5DdjvTp3vSVaeWLsY7gRn1FQ1FizH6GcU,40
|
|
3
|
+
comfi_fast_grnn_torch-0.0.3.dist-info/METADATA,sha256=TxS9LlHYJXnn2pQjt1IYBTqP-JpWM3mY5KI0szbekMc,1042
|
|
4
|
+
comfi_fast_grnn_torch-0.0.3.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
|
|
5
|
+
comfi_fast_grnn_torch-0.0.3.dist-info/top_level.txt,sha256=wEdR2DYmfjcMWO5alGNtgcwEhrPtTj6L3QICtKvX0M0,22
|
|
6
|
+
comfi_fast_grnn_torch-0.0.3.dist-info/RECORD,,
|
|
@@ -1,6 +0,0 @@
|
|
|
1
|
-
comfi_fast_grnn_torch/ComfiFastGRNN.py,sha256=uUX5h1ilLcicWtJt1BSPVGZmUOaeltWpjC6e4wq4pIE,10085
|
|
2
|
-
comfi_fast_grnn_torch/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
3
|
-
comfi_fast_grnn_torch-0.0.2.dist-info/METADATA,sha256=JJDjfZAB3QIpmI28altMMUxeo8Q5yJsIofFNuQa-3bk,1446
|
|
4
|
-
comfi_fast_grnn_torch-0.0.2.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
|
|
5
|
-
comfi_fast_grnn_torch-0.0.2.dist-info/top_level.txt,sha256=wEdR2DYmfjcMWO5alGNtgcwEhrPtTj6L3QICtKvX0M0,22
|
|
6
|
-
comfi_fast_grnn_torch-0.0.2.dist-info/RECORD,,
|
|
File without changes
|
{comfi_fast_grnn_torch-0.0.2.dist-info → comfi_fast_grnn_torch-0.0.3.dist-info}/top_level.txt
RENAMED
|
File without changes
|