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

updates

This commit is contained in:
Glenn Jocher 2019-07-16 18:59:46 +02:00
parent 153762dec0
commit c994963a84
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
cfg/yolov3-spp-1cls.cfg
View File

@ -3,8 +3,8 @@
# batch=1 # batch=1
# subdivisions=1 # subdivisions=1
# Training # Training
batch=64 batch=16
subdivisions=32 subdivisions=8
width=608 width=608
height=608 height=608
channels=3 channels=3

2
utils/torch_utils.py
View File

@ -26,7 +26,7 @@ def select_device(force_cpu=False):
c = 1024 ** 2 # bytes to MB c = 1024 ** 2 # bytes to MB
ng = torch.cuda.device_count() ng = torch.cuda.device_count()
x = [torch.cuda.get_device_properties(i) for i in range(ng)] x = [torch.cuda.get_device_properties(i) for i in range(ng)]
cuda_str = 'Using CUDA' + apex_str cuda_str = 'Using CUDA ' + apex_str
print("%sdevice0 _CudaDeviceProperties(name='%s', total_memory=%dMB)" % print("%sdevice0 _CudaDeviceProperties(name='%s', total_memory=%dMB)" %
(cuda_str, x[0].name, x[0].total_memory / c)) (cuda_str, x[0].name, x[0].total_memory / c))
if ng > 0: if ng > 0: