Jason.F issues

Repositories
Issues
Comments

Results 2 issues of


                                            Jason.F

Eerror of Centroid Triplet Loss (version1.2.0)

In version 1.2.0, Centroid Triplet Loss is unstable. When the batch size is small, the below error always occurs. If larger batch size, it is relatively be relieved. `File "/root/miniconda3/lib/python3.7/site-packages/pytorch_metric_learning/losses/base_metric_loss_function.py",...

bug

Install log

Before command "python setup.py install" you better command export CUDA_HOME="/usr/local/cuda/" , this is your cuda path.