Functions — github.com/bwconrad/decoder-denoising

CodeHub MCPcopy Create free account

hub / github.com/bwconrad/decoder-denoising / functions

Functions17 in github.com/bwconrad/decoder-denoising

⨍ Functions 17 ◇ Types & classes 4

↓ 2 callersMethoddenoise_step

(self, x, mode="train")

src/model.py:111

↓ 1 callersMethod__init__

Image dataset from directory Args: root: Path to directory transforms: Image augmentations

↓ 1 callersMethodadd_logger_args

src/pl_utils.py:13

↓ 1 callersMethodadd_noise

src/model.py:95

↓ 1 callersMethodget_loss_fn

(loss_type: str)

src/model.py:82

↓ 1 callersFunctioninit_logger

Initialize logger from arguments Args: args: parsed argument dictionary Returns: logger: initialized logger object

src/pl_utils.py:36

Method__getitem__

(self, *args: Any, **kwargs: Any)

src/pl_utils.py:9

Basic data module Args: root: Path to image directory size: Size of resized image crop: Size of image cro

Decoder Denoising Pretraining Model Args: lr: Learning rate optimizer: Name of optimizer (adam | adamw | sgd)

src/model.py:10

src/data.py:103

Methodconfigure_optimizers

src/model.py:140

(self, stage="fit")

Methodtrain_dataloader

Methodtraining_step

src/model.py:129

Methodval_dataloader

Methodvalidation_step

src/model.py:137