Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
29 changes: 28 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,12 @@ git clone https://github.com/mims-harvard/CLEF
cd CLEF
```

### :two: Set Up Environment
### :two: File and Folder Configurations
Step 2a) Inside `CLEF/project_config.py`, change `PROJECT_DIR = [insert your own project directory path]`.

Step 2b) Create an empty `CLEF/results` folder.

### :three: Set Up Environment

This codebase leverages Python, Pytorch, etc. To create an environment with all of the required packages, please ensure that [conda](https://docs.conda.io/projects/conda/en/latest/user-guide/install/index.html) is installed and then execute the commands:

Expand All @@ -44,6 +49,28 @@ conda env create -f environment.yml
conda activate clef_env
```

Please note that the environment file is for Linux environments and not compatible with macOS. If you're using a cluster, it's recommended to run `conda env create -f environment.yml` in an interactive node first before you submit your job script. Inside your job script, include the following before you run the training file:

```
source ~/.bashrc
conda activate clef_env
```

### :four: Set Up Wandb
Step 4a) Inside `CLEF/model/train.py`, replace the entity in the wandb.innit function to be your Wandb team name.

```
wandb.init(config = hparams_raw, project = "clef_project", entity = "[insert your Wandb team name]")
```

To find your Wandb team name: log into wandb → go to user settings → look under 'Default Team' to find your team location.

Step 4b) There are two common ways to log into wandb
1. Run `wandb login` in your terminal
2. Inside your SLURM job script, include `export WANDB_API_KEY = [your API key]` before you run the training file.



## Additional Resources

- [Data](https://doi.org/10.6084/m9.figshare.31580707)
Expand Down