folivo/PINA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Layers and Models update PR

Browse Source 
* add residual block
* add test conv and residual block
* modify FFN kwargs
This commit is contained in:
Dario Coscia
2023-08-01 19:13:36 +02:00
committed by Nicola Demo
parent 8c16e27ae4
commit 15ecaacb7c
9 changed files with 191 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
pina/model/layers/__init__.py
View File

@@ -1,7 +1,7 @@
__all__ = [
'BaseContinuousConv',
'ContinuousConv'
'ContinuousConvBlock',
'ResidualBlock'
]
from .convolution import BaseContinuousConv
from .convolution_2d import ContinuousConv
from .convolution_2d import ContinuousConvBlock
from .residual import ResidualBlock