RepSurf
RepSurf copied to clipboard
您好,想问问单卡3090的超参数您试过没没啊
您好,想问问单卡3090的config超参数您试过没没啊,由于我没得4张显卡,所以特向您请教
你好, 我在3090上尝试过,可以跑的。你可以尝试改这一行去实现单卡的运行。感谢你的关注。
Hi, I have tried the implementation on 3090, and it works as well. You could try changing this line in the script for segmentation, and so as for classification.
你好, 我在3090上尝试过,可以跑的。你可以尝试改这一行去实现单卡的运行。感谢你的关注。
Hi, I have tried the implementation on 3090, and it works as well. You could try changing this line in the script for segmentation, and so as for classification.
······················································································································································································· Following your tips, I successfully run it, I am running the ISPRS dataset, but the single card 3090 does not run very well, could you tell me what are the key parameters of this code? The following are my configuration parameters, thanks

I am not familiar with ISPRS dataset, but I think you can follow some common protocals: to scale the batch size by #gpus, e.g., 8 for 4 gpus and 2 for 1 gpu, and to scale the learning rate by a scaled batch size, e.g., 0.006 for 2, and 0.012 for 4. You can also tune the hyperparameters of augmentation and voxel size, which mainly depends on data, but it takes time and I am not 100% sure that it may be useful. Besides, more complete input and more input points could be benenficial (the best option is to set voxel_max to None), but take care of the OOM issue.