Update README.md
Browse files
README.md
CHANGED
@@ -31,8 +31,8 @@ from huggingface_hub import hf_hub_download
|
|
31 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
32 |
print(f'Using device: {device}')
|
33 |
|
34 |
-
# Define
|
35 |
-
username = "Vijayendra"
|
36 |
model_name_epoch_125 = "QST-CIFAR10-Epoch125"
|
37 |
model_name_best = "QST-CIFAR10-BestModel"
|
38 |
|
|
|
31 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
32 |
print(f'Using device: {device}')
|
33 |
|
34 |
+
# Define Hugging Face username and repository names
|
35 |
+
username = "Vijayendra"
|
36 |
model_name_epoch_125 = "QST-CIFAR10-Epoch125"
|
37 |
model_name_best = "QST-CIFAR10-BestModel"
|
38 |
|