Update experimental_temperature.py

This commit is contained in:
Extraltodeus
2024-05-24 17:31:20 +02:00
committed by GitHub
parent aae9050b0e
commit eb33ee5248
+1
View File
@@ -10,6 +10,7 @@ import os
def exists(val):
return val is not None
# better than a division by 0 hey
abs_mean = lambda x: torch.where(torch.isnan(x) | torch.isinf(x), torch.zeros_like(x), x).abs().mean()
class temperature_patcher():