CUDA_VISIBLE_DEVICES=0 python2 -u train_cell_base.py