Atlas icon indicating copy to clipboard operation
Atlas copied to clipboard

Parameters for fine-tuning

Open noahstier opened this issue 3 years ago • 0 comments

Can @zmurez, or anyone else, suggest a configuration for training / fine-tuning Atlas that can learn acceptably well for experimentation using only a single GPU? I wonder if zmurez discovered such a configuration during development.

I will try decreasing VOXEL_DIM_TRAIN, NUM_FRAMES_TRAIN, and RESNETS.DEPTH. However, the convergence time is so long that a hint here could save a lot of time.

Thanks

noahstier avatar Apr 21 '21 17:04 noahstier