Eval scripts fixed - Githubissues

KaLiMaLi555 / model_extraction_interiit

1 stars 0 forks source link

Eval scripts fixed #19

Open anime-sh opened 2 years ago

anime-sh commented 2 years ago

added two files, wait thora before merging this, colab has too many sessions to check this, however this looks okay and will mp work.

anime-sh commented 2 years ago

This now has neg sampling too

anime-sh commented 2 years ago

@shubheshanand ... had forgotten to push the neg sampling fix, I am creating a new tensor for this so I am not sure if that would cause memory blowup, ek baar dekhlo