digital-forensics-labs/Machine-Learning-Collection
1
0
Fork 0
mirror of https://github.com/aladdinpersson/Machine-Learning-Collection.git synced 2026-04-10 12:33:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

cyclegan, progan

Browse Source
This commit is contained in:
Aladdin Persson
2021-03-11 15:50:44 +01:00
parent 91b1fd156c
commit 2c53205f12
27 changed files with 276 additions and 238 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ML/Pytorch/object_detection/YOLOv3/utils.py
View File

@@ -380,8 +380,6 @@ def check_class_accuracy(model, loader, threshold):
tot_obj, correct_obj = 0, 0
for idx, (x, y) in enumerate(tqdm(loader)):
if idx == 100:
break
x = x.to(config.DEVICE)
with torch.no_grad():
out = model(x)