File size: 178 Bytes
25f71fa
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
#!/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