diff --git a/README.md b/README.md
index bece76e4..019c12bf 100644
--- a/README.md
+++ b/README.md
@@ -136,7 +136,7 @@ If our work is useful for your research, please consider citing:
### License
-This project is licensed under S-Lab License 1.0. Redistribution and use for non-commercial purposes should follow this license.
+This project is licensed under NTU S-Lab License 1.0. Redistribution and use should follow this license.
### Acknowledgement
diff --git a/basicsr/archs/vqgan_arch.py b/basicsr/archs/vqgan_arch.py
index f6dfcf4c..5ac69263 100644
--- a/basicsr/archs/vqgan_arch.py
+++ b/basicsr/archs/vqgan_arch.py
@@ -41,10 +41,10 @@ def forward(self, z):
mean_distance = torch.mean(d)
# find closest encodings
- # min_encoding_indices = torch.argmin(d, dim=1).unsqueeze(1)
- min_encoding_scores, min_encoding_indices = torch.topk(d, 1, dim=1, largest=False)
+ min_encoding_indices = torch.argmin(d, dim=1).unsqueeze(1)
+ # min_encoding_scores, min_encoding_indices = torch.topk(d, 1, dim=1, largest=False)
# [0-1], higher score, higher confidence
- min_encoding_scores = torch.exp(-min_encoding_scores/10)
+ # min_encoding_scores = torch.exp(-min_encoding_scores/10)
min_encodings = torch.zeros(min_encoding_indices.shape[0], self.codebook_size).to(z)
min_encodings.scatter_(1, min_encoding_indices, 1)
@@ -66,7 +66,6 @@ def forward(self, z):
"perplexity": perplexity,
"min_encodings": min_encodings,
"min_encoding_indices": min_encoding_indices,
- "min_encoding_scores": min_encoding_scores,
"mean_distance": mean_distance
}
diff --git a/facelib/detection/yolov5face/face_detector.py b/facelib/detection/yolov5face/face_detector.py
index 0103411e..79fdba0c 100644
--- a/facelib/detection/yolov5face/face_detector.py
+++ b/facelib/detection/yolov5face/face_detector.py
@@ -17,7 +17,7 @@
scale_coords_landmarks,
)
-IS_HIGH_VERSION = tuple(map(int, torch.__version__.split('+')[0].split('.')[:3])) >= (1, 9, 0)
+IS_HIGH_VERSION = tuple(map(int, torch.__version__.split('+')[0].split('.')[:2])) >= (1, 9, 0)
def isListempty(inList):