mirrors
/
DCL
mirror of https://github.com/JDAI-CV/DCL.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Update eval_model.py

master
SaiYiKi 2019-07-23 23:18:05 +08:00 committed by GitHub
parent 6d78afc333
commit 6203db874e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
utils/eval_model.py
View File

@ -17,7 +17,7 @@ import pdb
def dt():
return datetime.datetime.now().strftime("%Y-%m-%d-%H_%M_%S")
def eval_turn(model, data_loader, val_version, epoch_num, log_file):
def eval_turn(Config, model, data_loader, val_version, epoch_num, log_file):
model.train(False)
@ -50,7 +50,7 @@ def eval_turn(model, data_loader, val_version, epoch_num, log_file):
val_loss_recorder.update(loss)
val_celoss_recorder.update(ce_loss)
if outputs[1].size(1) != 2:
if Config.use_dcl and Config.cls_2xmul:
outputs_pred = outputs[0] + outputs[1][:,0:num_cls] + outputs[1][:,num_cls:2*num_cls]
else:
outputs_pred = outputs[0]