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

Backpropagation and fix test for OrthogonalBlock

Browse Source 
Co-authored-by: Dario Coscia <dariocos99@gmail.com>
    Co-authored-by: Gabriele Codega <gcodega@pascal.maths.sissa.it>
This commit is contained in:
Dario Coscia
2024-09-03 16:23:14 +02:00
committed by Nicola Demo
parent 59fc19798f
commit eea0cc0833
3 changed files with 125 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -9,6 +9,7 @@ __all__ = [
"FourierBlock2D",
"FourierBlock3D",
"PODBlock",
"OrthogonalBlock",
"PeriodicBoundaryEmbedding",
"FourierFeatureEmbedding",
"AVNOBlock",
@@ -25,6 +26,7 @@ from .spectral import (
)
from .fourier import FourierBlock1D, FourierBlock2D, FourierBlock3D
from .pod import PODBlock
from .orthogonal import OrthogonalBlock
from .embedding import PeriodicBoundaryEmbedding, FourierFeatureEmbedding
from .avno_layer import AVNOBlock
from .lowrank_layer import LowRankBlock