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

View File

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

View File

@ -26,7 +26,7 @@ def select_device(force_cpu=False):
c = 1024 ** 2 # bytes to MB
ng = torch.cuda.device_count()
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)" %
(cuda_str, x[0].name, x[0].total_memory / c))
if ng > 0: