search

hugoycj / torch-bakedsdf

An unofficial pytorch implementation of BakedSDF
https://bakedsdf.github.io/
MIT License
187 stars 8 forks source link

Fix validation loader #1

Closed sanskar107 closed 1 year ago

sanskar107 commented 1 year ago

Training crashes during validation loop as dataset.all_images are on cpu, while index is on cuda.