qinglew / PCN-PyTorch

Implementation of PCN(Point Completion Network) in PyTorch.
141 stars 33 forks source link

Difficulty reproducing training loss #32

Closed plebbyd closed 7 months ago

plebbyd commented 10 months ago

I'm using a modified version of the PCN model in order to train an AutoEncoder that essentially re-constructs the input:

The best loss (same value as CD) I am able to get is ~ 0.06. My question is: am I doing something wrong or missing a key component?

I would think that I should be able to achieve losses at least less than 1.Nx as this repository (10e-3). I am using a 3090 Ti with batch sizes anywhere from 256 to 1024.