Index

_

__init__() (fit_a_nef.initializers.InitModel method) (fit_a_nef.initializers.MetaLearnedInit method) (fit_a_nef.initializers.RandomInit method) (fit_a_nef.initializers.SharedInit method) (fit_a_nef.nef.mfn.FourierFilter method) (fit_a_nef.nef.mfn.FourierNet method) (fit_a_nef.nef.mfn.GaborFilter method) (fit_a_nef.nef.mfn.GaborNet method) (fit_a_nef.nef.mfn.MFNBase method) (fit_a_nef.nef.mlp.MLP method) (fit_a_nef.nef.rffnet.Layer method) (fit_a_nef.nef.rffnet.RFF method) (fit_a_nef.nef.rffnet.RFFNet method) (fit_a_nef.nef.siren.SIREN method) (fit_a_nef.nef.siren.SirenLayer method) (fit_a_nef.tasks.image.SignalImageTrainer method) (fit_a_nef.tasks.shape.SignalShapeTrainer method) (fit_a_nef.trainer.SignalTrainer method) (fit_a_nef.utils.TrainState method)

A

apply_model() (fit_a_nef.trainer.SignalTrainer method)

C

clean_up() (fit_a_nef.tasks.image.SignalImageTrainer method) (fit_a_nef.trainer.SignalTrainer method) compile() (fit_a_nef.tasks.shape.SignalShapeTrainer method) (fit_a_nef.trainer.SignalTrainer method)

create_functions() (fit_a_nef.trainer.SignalTrainer method) create_train_model() (fit_a_nef.trainer.SignalTrainer method) custom_uniform() (in module fit_a_nef.nef.utils)

F

fast_train_model() (fit_a_nef.tasks.shape.SignalShapeTrainer method) (fit_a_nef.trainer.SignalTrainer method) fit_a_nef module fit_a_nef.initializers module fit_a_nef.metrics module fit_a_nef.nef module fit_a_nef.nef.mfn module fit_a_nef.nef.mlp module fit_a_nef.nef.rffnet module fit_a_nef.nef.siren module

fit_a_nef.nef.utils module fit_a_nef.tasks module fit_a_nef.tasks.image module fit_a_nef.tasks.shape module fit_a_nef.trainer module fit_a_nef.utils module flatten_dict() (in module fit_a_nef.utils) flatten_params() (in module fit_a_nef.utils) fourier_filter_linear_init() (in module fit_a_nef.nef.mfn) FourierFilter (class in fit_a_nef.nef.mfn) FourierNet (class in fit_a_nef.nef.mfn) FourierNet_key() (in module fit_a_nef.nef.mfn)

G

GaborFilter (class in fit_a_nef.nef.mfn) GaborNet (class in fit_a_nef.nef.mfn) GaborNet_key() (in module fit_a_nef.nef.mfn) gamma_initialization() (in module fit_a_nef.nef.mfn) gaussian_window() (in module fit_a_nef.nef.mfn) get_flat_params() (fit_a_nef.trainer.SignalTrainer method)

get_lr() (fit_a_nef.trainer.SignalTrainer method) get_meta_init() (in module fit_a_nef.utils) get_nef() (in module fit_a_nef.utils) get_optimizer() (in module fit_a_nef.utils) get_params() (fit_a_nef.trainer.SignalTrainer method) get_scheduler() (in module fit_a_nef.utils)

I

init_model() (fit_a_nef.tasks.image.SignalImageTrainer method) (fit_a_nef.tasks.shape.SignalShapeTrainer method) (fit_a_nef.trainer.SignalTrainer method)

InitModel (class in fit_a_nef.initializers) iou() (in module fit_a_nef.metrics)

K

kmeans() (in module fit_a_nef.metrics)

L

Layer (class in fit_a_nef.nef.rffnet)

load() (fit_a_nef.trainer.SignalTrainer method) load_model_cfg() (in module fit_a_nef.nef.utils)

M

mae() (fit_a_nef.tasks.image.SignalImageTrainer method) (in module fit_a_nef.metrics) MetaLearnedInit (class in fit_a_nef.initializers) MFNBase (class in fit_a_nef.nef.mfn) MLP (class in fit_a_nef.nef.mlp) MLP_key() (in module fit_a_nef.nef.mlp) module fit_a_nef fit_a_nef.initializers fit_a_nef.metrics fit_a_nef.nef fit_a_nef.nef.mfn fit_a_nef.nef.mlp fit_a_nef.nef.rffnet fit_a_nef.nef.siren fit_a_nef.nef.utils fit_a_nef.tasks fit_a_nef.tasks.image fit_a_nef.tasks.shape fit_a_nef.trainer fit_a_nef.utils

mse() (fit_a_nef.tasks.image.SignalImageTrainer method) (in module fit_a_nef.metrics)

N

nmi() (in module fit_a_nef.metrics)

P

process_batch() (fit_a_nef.tasks.image.SignalImageTrainer method) (fit_a_nef.trainer.SignalTrainer method)

psnr() (fit_a_nef.tasks.image.SignalImageTrainer method) (in module fit_a_nef.metrics)

R

RandomInit (class in fit_a_nef.initializers) replace() (fit_a_nef.utils.TrainState method) RFF (class in fit_a_nef.nef.rffnet)

RFFNet (class in fit_a_nef.nef.rffnet) RFFNet_key() (in module fit_a_nef.nef.rffnet) rmse() (in module fit_a_nef.metrics)

S

save() (fit_a_nef.trainer.SignalTrainer method) setup() (fit_a_nef.nef.mfn.FourierFilter method) (fit_a_nef.nef.mfn.FourierNet method) (fit_a_nef.nef.mfn.GaborFilter method) (fit_a_nef.nef.mfn.GaborNet method) (fit_a_nef.nef.mfn.MFNBase method) (fit_a_nef.nef.rffnet.Layer method) (fit_a_nef.nef.rffnet.RFF method) (fit_a_nef.nef.rffnet.RFFNet method) (fit_a_nef.nef.siren.SIREN method) (fit_a_nef.nef.siren.SirenLayer method) SharedInit (class in fit_a_nef.initializers)

SignalImageTrainer (class in fit_a_nef.tasks.image) SignalShapeTrainer (class in fit_a_nef.tasks.shape) SignalTrainer (class in fit_a_nef.trainer) simple_uniform() (in module fit_a_nef.nef.mfn) simse() (fit_a_nef.tasks.image.SignalImageTrainer method) (in module fit_a_nef.metrics) SIREN (class in fit_a_nef.nef.siren) SIREN_key() (in module fit_a_nef.nef.siren) SirenLayer (class in fit_a_nef.nef.siren) ssim() (fit_a_nef.tasks.image.SignalImageTrainer method) (in module fit_a_nef.metrics) stack_cut_tensor() (in module fit_a_nef.tasks.shape)

T

train_to_target_psnr() (fit_a_nef.tasks.image.SignalImageTrainer method)

TrainState (class in fit_a_nef.utils)

U

unflatten_dict() (in module fit_a_nef.utils)

unflatten_params() (in module fit_a_nef.utils)

V

validation_psnr() (fit_a_nef.tasks.image.SignalImageTrainer method) verbose_train_model() (fit_a_nef.tasks.image.SignalImageTrainer method) (fit_a_nef.tasks.shape.SignalShapeTrainer method) (fit_a_nef.trainer.SignalTrainer method)