Spaces:

Muthukamalan
/

DogBreedsClassifier

Sleeping

Muthukamalan commited on Nov 25, 2024

Commit

319ddab

verified ·

1 Parent(s): 4e376e0

Upload folder using huggingface_hub

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ torch.set_float32_matmul_precision("medium")
 # device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
 device = torch.device("cpu")
 torch.set_default_device(device=device)
-torch.autocast(enabled=True, dtype="float16", device_type="cuda")
 pl.seed_everything(123, workers=True)

 # device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
 device = torch.device("cpu")
 torch.set_default_device(device=device)
+# torch.autocast(enabled=True, dtype="float16", device_type="cuda")
 pl.seed_everything(123, workers=True)