Rename files
This commit is contained in:
committed by
Nicola Demo
parent
886bd23fdb
commit
ff43a7492b
@@ -25,11 +25,11 @@ from .spectral import (
|
||||
SpectralConvBlock2D,
|
||||
SpectralConvBlock3D,
|
||||
)
|
||||
from .fourier import FourierBlock1D, FourierBlock2D, FourierBlock3D
|
||||
from .pod import PODBlock
|
||||
from .fourier_block import FourierBlock1D, FourierBlock2D, FourierBlock3D
|
||||
from .pod_block import PODBlock
|
||||
from .orthogonal import OrthogonalBlock
|
||||
from .embedding import PeriodicBoundaryEmbedding, FourierFeatureEmbedding
|
||||
from .avno_layer import AVNOBlock
|
||||
from .lowrank_layer import LowRankBlock
|
||||
from .rbf_layer import RBFBlock
|
||||
from .average_neural_operator_block import AVNOBlock
|
||||
from .lowrank_block import LowRankBlock
|
||||
from .rbf_block import RBFBlock
|
||||
from .gno_block import GNOBlock
|
||||
|
||||
@@ -1,7 +1,7 @@
|
||||
""" Module for Averaging Neural Operator Layer class. """
|
||||
|
||||
from torch import nn, mean
|
||||
from pina.utils import check_consistency
|
||||
from ...utils import check_consistency
|
||||
|
||||
|
||||
class AVNOBlock(nn.Module):
|
||||
@@ -2,7 +2,7 @@ import torch
|
||||
import torch.nn as nn
|
||||
from ...utils import check_consistency
|
||||
|
||||
from pina.model.block import (
|
||||
from . import (
|
||||
SpectralConvBlock1D,
|
||||
SpectralConvBlock2D,
|
||||
SpectralConvBlock3D,
|
||||
@@ -1,11 +0,0 @@
|
||||
""" Module for Averaging Neural Operator Layer class. """
|
||||
|
||||
from torch import nn, mean
|
||||
from torch_geometric.nn import MessagePassing, InstanceNorm, radius_graph
|
||||
|
||||
from pina.utils import check_consistency
|
||||
|
||||
|
||||
class MessagePassingBlock(nn.Module):
|
||||
|
||||
|
||||
Reference in New Issue
Block a user