Skip to content

Two experiments (#1 GAN image generation; #2 image classification with Resnet; both in PyTorch) to test differences in training times between free Colab GPUs and mobile/desktop GPUs.

Notifications You must be signed in to change notification settings

pchaberski/GPU_tests

Repository files navigation

Google Colab remote GPUs vs. mobile GPU performance

Two experiments to test differences in training times between free Colab GPUs and mobile/desktop GPUs.

GPUs tested

  • GeForce GTX 1050 Mobile (local notebook)
  • GeForce GTX 1060 Desktop (local desktop)
  • GeForce RTX 2060 Mobile (local notebook)
  • Tesla P4 (Colab free)
  • Tesla K80 (Colab free)
  • Telsa T4 (Colab free)
  • Tesla P100 (Colab free)

Experiment no. 1 definition

  • task: image generation using convolutional GAN
  • dataset: MNIST
  • framework used: PyTorch
  • 50 epochs
  • experiment notebook

Experiment no. 1 model results

Sample of generated images and losses:

Experiment no. 1 training times comparison

Experiment no. 2 definition

  • task: image classification using Resnet-18
  • dataset: CIFAR-100
  • framework used: PyTorch
  • 10 epochs
  • experiment notebook

Experiment no. 2 model results

Losses and accuracy:

Experiment no. 2 training times comparison

About

Two experiments (#1 GAN image generation; #2 image classification with Resnet; both in PyTorch) to test differences in training times between free Colab GPUs and mobile/desktop GPUs.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published