Speed Tests - ProGamerGov/pytorch-vgg GitHub Wiki
Using the timing scripts that I have created for Neural-Style-PT: https://gist.github.com/ProGamerGov/e5e94f8735c1f7991a3e4ad2289e0d01
Currently these are the GPUs that I have tested Neural-Style-PT with:
Tesla K80:
Command | Time |
---|---|
LBFGS nn | 117 seconds |
LBFGS cudnn | 124 seconds |
LBFGS cudnn autotune | 109 seconds |
Adam nn | 100 seconds |
Adam cudnn | 107 seconds |
Adam cudnn autotune | 91 seconds |
GRID K520:
Command | Time |
---|---|
LBFGS nn | 236 seconds |
LBFGS cudnn | 226 seconds |
LBFGS cudnn autotune | 226 seconds |
Adam nn | 209 seconds |
Adam cudnn | 200 seconds |
Adam cudnn autotune | 200 seconds |
GTX 1080:
Command | Time |
---|---|
LBFGS nn | 56 seconds |
LBFGS cudnn | 38 seconds |
LBFGS cudnn autotune | 40 seconds |
Adam nn | 40 seconds |
Adam cudnn | 23 seconds |
Adam cudnn autotune | 24 seconds |