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 <[email protected]>
# Modified by François Lagunas <[email protected]>
import time, torch