Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
lzhbrian committed Apr 9, 2019
1 parent 74ecbe0 commit 8d286e8
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion is_fid_pytorch.py
Original file line number Diff line number Diff line change
Expand Up @@ -246,7 +246,7 @@ def __calc_is(preds, n_split, return_each_score=False):
scores.append(entropy(pyx, py))
split_scores.append(np.exp(np.mean(scores)))
if n_split == 1 and return_each_score:
return np.mean(split_scores), np.std(split_scores), scores
return scores, 0
return np.mean(split_scores), np.std(split_scores)

@staticmethod
Expand Down

0 comments on commit 8d286e8

Please sign in to comment.