Update README.md

This commit is contained in:
bryandlee 2021-02-21 21:02:20 +09:00 committed by GitHub
parent 504504fbda
commit 8314798e04
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -28,4 +28,4 @@ python test.py --input_dir [image_folder_path]
<img src="./samples/compare/2.jpg" width="960"> &nbsp;
<img src="./samples/compare/3.jpg" width="960"> &nbsp;
**Note:** Tested on RTX3090 + PyTorch1.7.1 / Results from converted weights slightly different due to the [bilinear upsample issue](https://github.com/pytorch/pytorch/issues/10604)
**Note:** Training code not included / Tested on RTX3090 + PyTorch1.7.1 / Results from converted weights slightly different due to the [bilinear upsample issue](https://github.com/pytorch/pytorch/issues/10604)