about a problem in gpu_version

naturomics / CapsNet-Tensorflow

A Tensorflow implementation of CapsNet(Capsules Net) in paper Dynamic Routing Between Capsules

Apache License 2.0

3.8k stars 1.17k forks source link

about a problem in gpu_version #84

Open Questdream opened 4 years ago

Questdream commented 4 years ago

thanks your codes. I finished the normal version, but when I used the dist_version, it occurs a issue. The errors shows "AssertionError :assert not np.isnan(loss_value)" in distributed_train.py. in line 132. how's it going? can you help me ?