pytorchTutorial
pytorchTutorial copied to clipboard
PyTorch Tutorials from my YouTube channel
Please change the name of the file transformers to transforms.
Hi, i have an question for the .py line 141 : ``` class_labels.append(predicted) ``` why not ``` class_labels.append(labels) ``` the add_pr_curve() function should use the ground truth label?
Hi @python-engineer , on line 131, shouldn't we assign z again on CPU? z = z.to("cpu") before then calling z.numpy() I got following error without doing so: TypeError: can't convert...
Fix a typo
model was on graphic card and example_data was on cpu so was throwing error
Based on pytorch docs, the parameter "labels" of add_pr_curve() should be ground truth data. But in Line 141, class_labels append our own predictions. Therefore, I substitute "labels" for "predicted".
bug fix in gradient calculation
Add comment to have a FC layer of the same size as len(class_name)... it was commented on l:138 but not in l:172
Where is the dataset data/hymenoptera_data in transfer learning tutorial from? It seems that in previous tutorials it was not mentioned how this dataset was obtained.