PRBonn / deep-point-map-compression

MIT License
94 stars 28 forks source link

transfer to a new dataset #11

Closed tuning12 closed 11 months ago

tuning12 commented 1 year ago

First thank you for your great work. I have a question about using your model on my dataset. My trainingset, validset and testingset is three .pkl files with sizes of [B, N, 3] which is normalized already. Could you tell me how to transform my data to the form which could be trained? Thank you so much.

louis-wiesmann commented 11 months ago

This version can only handle batch size 1. But you could accumulate gradients for B times and update the weights only after this. This should be equivalent.