zyxu1996 / Efficient-Transformer

Online !!! Application of an efficient transformer improved based on Swin transformer on remote sensing segmentation
Apache License 2.0
72 stars 7 forks source link

Model accuracy #8

Open PeterLuoCoder opened 1 year ago

PeterLuoCoder commented 1 year ago

In your paper, Table 10 shows that models like FCN, Unet, and DANet did not achieve the results as presented in the table, even after training and testing them twice. For example, in the case of DANet, I trained it using a single NVIDIA 3090 GPU. I made some adjustments such as modifying your multi-GPU setup to a single GPU and setting the batch size to 8 (both 4 and 8 gave similar results). Everything else remained the same. Can you provide any suggestions? Does using a single GPU affect the accuracy, or are there any other crucial parameter settings that I might have overlooked? Please advise. Thank you. bf6c6ee18d8e1049c8a578c471be217

zyxu1996 commented 1 year ago

You should use lr=0.0003, bs=16, img_size=512