Atlas
Atlas copied to clipboard
Parameters for fine-tuning
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