weecology / DeepTreeAttention

Implementation of Hang et al. 2020 "Hyperspectral Image Classification with Attention Aided CNNs" for tree species prediction
MIT License
119 stars 38 forks source link

weight decay from torchvisions blogpost #129

Closed bw4sz closed 2 years ago

bw4sz commented 2 years ago

Just checking out a few of these improvements. https://pytorch.org/blog/how-to-train-state-of-the-art-models-using-torchvision-latest-primitives/