Skip to content

Latest commit

 

History

History
20 lines (12 loc) · 526 Bytes

File metadata and controls

20 lines (12 loc) · 526 Bytes

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