Commit 23f12127 authored by lishen's avatar lishen

[fix]

parent 6cb1b7ad
...@@ -12,7 +12,7 @@ import torch.backends.cudnn as cudnn ...@@ -12,7 +12,7 @@ import torch.backends.cudnn as cudnn
import re import re
from utils import * from utils import *
# os.environ["CUDA_VISIBLE_DEVICES"] = "0" os.environ["CUDA_VISIBLE_DEVICES"] = "0,1,2,3,4"
def parse_option(): def parse_option():
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment