Skip to content

Commit 9a7c76f

Browse files
authored
Update README.md
1 parent 6a526d5 commit 9a7c76f

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

README.md

+6-3
Original file line numberDiff line numberDiff line change
@@ -9,13 +9,15 @@ Project with Neural Networks for Image Classification based on PyTorch.
99

1010
## 🚀 Train
1111

12-
### 1. Install all necessary libs:
12+
### 1. Install all necessary libs:
1313
```sh
1414
pip3 install -r requirements.txt
1515
```
16+
1617
Install torch.
1718
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.
1819
Otherwise, the processor will be used.
20+
1921
-----
2022
### 2. Dataset structure
2123
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:
5052
* `callbacks` - pytorch-lightning callbacks for your train
5153
* `arch` (in `model` params) - backbone name supported by [timm](https://github.com/huggingface/pytorch-image-models)
5254

53-
<details>
54-
<summary>Supported backbones</summary>
5555
To get all supported backbones names use:
5656
```
5757
timm.list_models()
5858
```
59+
<details>
60+
<summary>Supported backbones</summary>
61+
5962
| backbone | backbone | backbone | backbone | backbone | backbone | backbone | backbone | backbone |
6063
|----------|----------|----------|----------|----------|----------|----------|----------|----------|
6164
|botnet50ts_256|caformer_b36|caformer_m36|caformer_s18|caformer_s36|cait_m36_384|cait_m48_448|cait_s24_224|cait_s24_384|

0 commit comments

Comments
 (0)