AstroHack - PyTorch Example.

A tutorial that demonstrates how to predict galaxy M/L ratios using a simple convolutional neural network with the Pytorch. Sample notebook written and tested using Python 3.6.

How to use.

Local environment.

This notebook prefer to have local Python environment configured and GPU. To run on CPU changes configuration and create a model on CPU.

1. Clone this repository to your local environment.
2. Install and setup Jupyter notebook and dependencies.
3. Download Astrohack dataset and copy to ./data/ folder.
4. Run Astrohack_Pytorch.ipynb notebook.