Skip to content
Permalink
Browse files

update AutoDL Design

  • Loading branch information...
alexqdh committed Apr 15, 2019
1 parent 9f08071 commit f02208f664bac6fd70288d9198f631eb1b7e5c92
Showing with 2 additions and 2 deletions.
  1. +1 −1 AutoDL Design/autodl.py
  2. +1 −1 AutoDL Design/run.sh
@@ -121,7 +121,7 @@ def supervisor(self, mid):
--early_stop=20 --data_path=./cifar/pickle-cifar-10
"""
tokens, adjvec = utils.load_action(mid, self.log_dir)
cmd = ("CUDA_VISIBLE_DEVICES=2 python -u inception_train/train.py \
cmd = ("CUDA_VISIBLE_DEVICES=1 python -u inception_train/train.py \
--mid=%d --early_stop=%d --logdir=%s --data_path=%s --chunk_size=%d") % \
(mid, self.early_stop, self.log_dir, self.data_path, self.chunk_size)
print("cmd:{}".format(cmd))
@@ -1,4 +1,4 @@
export FLAGS_fraction_of_gpu_memory_to_use=0.98
export FLAGS_eager_delete_tensor_gb=0.0
export FLAGS_fast_eager_deletion_mode=1
CUDA_VISIBLE_DEVICES=1 python -u main.py > a.log 2>&1
CUDA_VISIBLE_DEVICES=0 python -u main.py > main.log 2>&1 &

0 comments on commit f02208f

Please sign in to comment.
You can’t perform that action at this time.