Skip to content

Instantly share code, notes, and snippets.

View madlag's full-sized avatar

François Lagunas madlag

  • Paris, France
View GitHub Profile
@madlag
madlag / speed2.py
Created September 30, 2020 17:35
Pytorch CUDA speed test for various data types, with and without AMP
#!/usr/bin/env python
# Any copyright is dedicated to the Public Domain.
# https://creativecommons.org/publicdomain/zero/1.0/
# Written by Francois Fleuret <francois@fleuret.org>
# Modified by François Lagunas <francois.lagunas@m4x.org>
import time, torch