DECIDIA-code / train.sh
lixiangchun
first commit
25f71fa
#!/bin/bash
python train.py \
--sequence_embedding sequence_embedding \
--input_dir data \
--num_classes 2 \
--device 'cuda:0' \
--num_train_patients 200 \
--num_sequences 200