support finetune with paired data

2021-08-27 16:14:48 +08:00
parent 194c2c14b3
commit f5ccd64ce5
11 changed files with 426 additions and 7 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,4 +5,5 @@ numpy
 opencv-python
 Pillow
 torch>=1.7
+torchvision
 tqdm