pmelchior / scarlet2

Scarlet, all new and shiny
MIT License
15 stars 4 forks source link

Update nn.py #37

Closed SampsonML closed 8 months ago

SampsonML commented 8 months ago

TempScore class needs to be derived class of ScorePrior with new implementation and galaxygrad versions. Single line fix, tested and working.

Example usage:

temp = 0.02
prior = nn.TempScore(model=HSC_ScoreNet64, temp=temp)
SampsonML commented 8 months ago

Closed, including this one liner in #42