search

cvqluu / Angular-Penalty-Softmax-Losses-Pytorch

Angular penalty loss functions in Pytorch (ArcFace, SphereFace, Additive Margin, CosFace)
MIT License
481 stars 92 forks source link

draw sphere fail #5

Open carr123 opened 4 years ago

carr123 commented 4 years ago

File "D:/project/pydemo/Angular-Penalty-Softmax-Losses-Pytorch/train_fMNIST.py", line 47, in main plot(am_embeds, am_labels, fig_path='./figs/{}.png'.format(loss_type)) File "D:\project\pydemo\Angular-Penalty-Softmax-Losses-Pytorch\plotting.py", line 28, in plot ax.set_aspect("equal") File "D:\Program Files\Python37\lib\site-packages\matplotlib\axes_base.py", line 1264, in set_aspect 'It is not currently possible to manually set the aspect ' NotImplementedError: It is not currently possible to manually set the aspect on 3D axes

Huong-nt commented 4 years ago

Try to remove line 28 in plotting.py: ax.set_aspect("equal")