Skip to content

Commit 97f074e

Browse files
authored
Update README.md
1 parent 55e38d0 commit 97f074e

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

README.md

+3-1
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# UNet-based-Denoising-Autoencoder
1+
# UNet-based-Denoising-Autoencoder-In-PyTorch
22
Cleaning printed text using Denoising Autoencoder based on UNet architecture in PyTorch
33

44
## Requirements
@@ -21,13 +21,15 @@ Set the desired values of **lr**, **epochs** and **batch_size** in *config.py*
2121
* ### Start Training
2222
In *config.py*,
2323
* Set **resume** to False
24+
2425
```
2526
python train.py
2627
```
2728
* ### Resume Training
2829
In *config.py*,
2930
* set **resume** to True and
3031
* set **ckpt** to the path of the model to be loaded, i.e. ckpt = 'model02.pth'
32+
3133
```
3234
python train.py
3335
```

0 commit comments

Comments
 (0)