#!/usr/bin/env bash
export CUDA_VISIBLE_DEVICES=0
export PYTHONPATH=${PYTHONPATH}:`pwd`
config_path='freenet.freenet_1_0_salinas'
model_dir='./log/salinas/freenet/1.0_poly'
python train.py \
--config_path=${config_path} \
--model_dir=${model_dir} \
train.save_ckpt_interval_epoch 9999
# part of train.py
if __name__ == '__main__':
torch.backends.cudnn.benchmark = True
args = train.parser.parse_args()
SEED = 2333
torch.manual_seed(SEED)
torch.cuda.manual_seed(SEED)
train.run(config_path=args.config_path,
model_dir=args.model_dir,
cpu_mode=args.cpu,
after_construct_launcher_callbacks=[register_evaluate_fn],
opts=args.opts)
Моя программа показана выше, но вы не можете щелкнуть Debug
в файле. sh в Pycharm. Как я могу изменить мою программу? Заранее спасибо!