OrgLion-ML
/
darknet
mirror of https://github.com/AlexeyAB/darknet.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Compile fix

This commit is contained in:
AlexeyAB 2020-06-08 16:35:24 +03:00
parent 7054d0b27e
commit 462980d2d9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/blas.h
View File

@ -155,7 +155,7 @@ void reduce_and_expand_array_gpu(const float *src_gpu, float *dst_gpu, int size,
void expand_array_gpu(const float *src_gpu, float *dst_gpu, int size, int groups); void expand_array_gpu(const float *src_gpu, float *dst_gpu, int size, int groups);
float cosine_similarity(float *A, float *B, unsigned int feature_size); float cosine_similarity(float *A, float *B, unsigned int feature_size);
float P_constrastive(int i, int l, int num_of_samples, float **z, unsigned int feature_size, float temperature); float P_constrastive(int i, int l, int *labels, int num_of_samples, float **z, unsigned int feature_size, float temperature);
void grad_contrastive_loss_positive(int i, int *labels, int num_of_samples, float **z, unsigned int feature_size, float temperature, float *delta); void grad_contrastive_loss_positive(int i, int *labels, int num_of_samples, float **z, unsigned int feature_size, float temperature, float *delta);
void grad_contrastive_loss_negative(int i, int *labels, int num_of_samples, float **z, unsigned int feature_size, float temperature, float *delta); void grad_contrastive_loss_negative(int i, int *labels, int num_of_samples, float **z, unsigned int feature_size, float temperature, float *delta);