- Add fft-shifting images
- Add details how metrics are computed
- Add details how the x-data is scaled?
- Fill in values for numbers of epochs in the paper
- Add difference histograms
- Correct training description
- AdamW instead of Adam, with weight decay 1e-5
- Warm-up for 3000 iterations or equivalent to 1 epoch
- Re-train models with corrected random number generation
Code presentation:
- Add data into the package (see C1 lesson)
- Add model as well?
- Make a package (see C1 lesson)
- Update pyproject.toml
- Add CLI version