/Galaxy_Zoo

NYU Computer Vision 2018 project

Primary LanguagePython

Galaxy_Zoo

Project of Computer Vision 2018 NYU course. We have achieved SOTA performance by RMSE 0.7484. Report please refer to here. Work done in collaboration with Lei Chen.

To run on HPC cluster

  1. Create conda environment, but you may not be supposed to run it with single core cpu.
conda env create -f requirements.yaml
  1. Run the training phase
mkdir -p hpc_outputs
mkdir -p models
sbatch shell/hpcrun.sb
  1. To generate test result to submit
sbatch shell/hpceval.sb