NJUVISION / PCGCv1

Point Cloud Geometry Compression
Apache License 2.0
42 stars 11 forks source link

A question about my Pytorch version #6

Closed tengqi200 closed 2 years ago

tengqi200 commented 2 years ago

I try to re-implement this demo in Pytorch 1.12 using a 2-byte floating format. And then, I found that the elapsed time of the training process is too much even if only "points64_part1" is involved. 

My experiment is performed on two GTX2080ti. Or I need a SSD.

It seems that I don't need to train the model with a complete dataset every epoch, just like your code.