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>.

Name		Name	Last commit message	Last commit date
Latest commit History 4 Commits
args		args
bash_scripts/classifier		bash_scripts/classifier
classifier		classifier
datasets		datasets
logger		logger
model		model
preprocessing		preprocessing
.DS_Store		.DS_Store
README.md		README.md
environment.yml		environment.yml

Provide feedback