openclimatefix / PVNet

PVnet main repo

MIT License

17 stars 4 forks source link

Fix the horizon graph training on GPU #176

Closed dfulu closed 5 months ago

dfulu commented 5 months ago

Pull Request

Horizon loss graph fails when training on GPU since the loss values are GPU tensors. This fixes the problem by converting to numpy arrays when adding to store