This is a PyTorch implementation of the forward-forward algorithm. This project is inspired by an article below.
- Geoffrey Hinton, The Forward-Forward Algorithm: Some Preliminary Investigations, arXiv.2212.13345.
- PyTorch
- TorchVision
- NumPy
- Pandas
- Just run the command below
python main.py
- Then, you can get training results and f1-scores.
- Additionaly, main_notebook.ipynb does same work as main.py
- I refered to great works such Keras implementation (at entire pipeline) and mohammadpz's work (at prediction task)