X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=main.py;h=71026c59640be3752f3a311dc6f58cbcf49b11d1;hb=5d46a9bd7d032d90ef4c4b38ac3c9b5b66526527;hp=901b1d0529bb525b1cbca2b5c3bc91af7b12bf36;hpb=439c597d409c344283f8996f042daf79d3f24de2;p=picoclvr.git diff --git a/main.py b/main.py index 901b1d0..71026c5 100755 --- a/main.py +++ b/main.py @@ -5,9 +5,6 @@ # Written by Francois Fleuret -# torch.backends.cuda.matmul.allow_tf23 -# torch.autocast(torch.bfloat16) - import math, sys, argparse, time, tqdm, os import torch, torchvision