Run examples:

Use default parameters from config.py:

python main.py

Update default parameters from a command line:

python main.py --lr 0.0001 --batch_size 64

Update default parameters from a config file:

python main.py --config_path exp.yaml

TODO:

• Infinite dataloader
• Save Image and Intermediate metrics - check how PIM outside of the signal changes
• Try other, convolution like models
• Rewrite linear model with dot product instead of linear layer