search

brjathu / deepcaps

Official Implementation of "DeepCaps: Going Deeper with Capsule Networks" paper (CVPR 2019).
MIT License
151 stars 48 forks source link

fixed accuracy calculation #17

Closed Utkarsh87 closed 4 years ago

Utkarsh87 commented 4 years ago

Earlier the line p1 = np.sum().. had undefined variables. Changed it to stay consistent with the test function written in the train.py file.