#!/bin/bash
export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
function run_train () {
echo "training"
python3 train.py \
--data_path data/simple-examples/data/ \ --model_type small \
--use_gpu True \
#--init_from_pretrain_model models/0/params
}
run_train
