Please enable JavaScript.
Coggle requires JavaScript to display documents.
PyTorch Custom Datasets - Coggle Diagram
- Importing PyTorch and setting up device-agnostic code
-
- Model 0: TinyVGG without data augmentation
-
-
-
-
-
-
-
-
- Model 1: TinyVGG with Data Augmentation
-
-
-
-
-
- Become one with the data (data preparation)
-
- What should an ideal loss curve look like?
-
-
-
-
- Option 1: Loading Image Data Using ImageFolder
-
- Other forms of transforms (data augmentation - Thay đổi dữ liệu)
- Option 2: Loading Image Data with a Custom Dataset
-
-
-
-
- Make a prediction on a custom image
- Make a prediction on a custom image
-
-