Skip to content

Latest commit

 

History

History
15 lines (11 loc) · 372 Bytes

README.md

File metadata and controls

15 lines (11 loc) · 372 Bytes

brain-ct-classifier

CS231A Final Project Code

Run training on the adult dataset by running

bash_scripts/classifier/classifier_adult.sh

Run training on the adult dataset by running

bash_scripts/classifier/classifier_peds.sh

To conduct transfer learning, pass in the checkpoint path of the pretrained model using the arg: --ckpt_path <path to ckpt>.