Skip to content

GitLab

Unverified Commit 17fb08d6 authored by Srikumar Sastry's avatar Srikumar Sastry Committed by GitHub
Browse files

Import partial

parent f9e8ad18
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
torch_harmonics/examples/sfno/models/sfno.py
View file @ 17fb08d6
......@@ -36,6 +36,8 @@ from torch_harmonics import *
from .layers import *
from functools import partial
class SpectralFilterLayer(nn.Module):
"""
Fourier layer. Contains the convolution part of the FNO/SFNO
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment