neuro_morpho.cli

Command line interface for training and running models.

Functions

register_torch_transforms(→ None)	Register torch transforms to gin.
main(→ None)	Run the training and inference pipeline.

Module Contents

neuro_morpho.cli.register_torch_transforms() → None

Register torch transforms to gin.

This allows the user to configure torchvision transforms from the gin config file.

neuro_morpho.cli.main(config: str = 'unet.config.gin') → None

Run the training and inference pipeline.

Parameters:

config (str) – The path to the gin configuration file.