You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardexpand all lines: README.md
+6-3
Original file line number
Diff line number
Diff line change
@@ -9,13 +9,15 @@ Project with Neural Networks for Image Classification based on PyTorch.
9
9
10
10
## 🚀 Train
11
11
12
-
### 1. Install all necessary libs:
12
+
### 1. Install all necessary libs:
13
13
```sh
14
14
pip3 install -r requirements.txt
15
15
```
16
+
16
17
Install torch.
17
18
Note: if you are using a GPU, then you need to install CUDA and replace the torch version in `requirements.txt` with the GPU-enabled version.
18
19
Otherwise, the processor will be used.
20
+
19
21
-----
20
22
### 2. Dataset structure
21
23
Directory with 2 subdirectories: `tran_val` with the number of subdirectories equal to num classes and `test` with the number of subdirectories equal to num classes:
@@ -50,12 +52,13 @@ The most important:
50
52
*`callbacks` - pytorch-lightning callbacks for your train
51
53
*`arch` (in `model` params) - backbone name supported by [timm](https://github.com/huggingface/pytorch-image-models)
0 commit comments