Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2018-11-22 20:03:09 +01:00
parent 6f83c321c8
commit 887ab29c64
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -18,7 +18,7 @@ opt = parser.parse_args()
print(opt)
# Import test.py to get mAP after each epoch
sys.argv[1:] = [] # delete any command line arguments that might get picked up by test.py
sys.argv[1:] = [] # delete any train.py command-line arguments before they reach test.py
import test # must follow sys.argv[1:] = []
cuda = torch.cuda.is_available()