swa

swa copied to clipboard

Figure 3 plot

1

Hi, can you update the code for plotting Figure 3 in the paper?

byan19

SWA with Torchbearer

3

Hi! I am trying to implement SWA with Torchbearer but I'm having issues with the state etc. Could anybody point me in the direction of some resources to better understand...

AnabetsyR

SWA with distributed training

1

In case of distributed training, e.g. DDP, each gpu will only process a minibatch, and the bn statistics computed in each gpu are different. When SWA is adopted, we need...

milliema

Do you have a keras version?

12

xuzhang5788

make it compatible for pytorch latest version

1

Hi, just noticed this wonderful work. I have made this code available for latest PyTorch version by providing minor revisions. Please consider merging.

jizongFox

I see def conv_init(m): in the wide_resnet.py of models, but it just be defined, and it's not applyed. Is it should be used as net.apply(conv_init)?

DCC-lzhy

feat: Set up complete Python testing infrastructure with Poetry

2

# Set Up Python Testing Infrastructure ## Summary This PR establishes a complete testing infrastructure for the PyTorch SWA-Gaussian project using Poetry as the package manager and pytest as the...

llbbl