huawei-noah / AdderNet

Code for paper " AdderNet: Do We Really Need Multiplications in Deep Learning?"
BSD 3-Clause "New" or "Revised" License
954 stars 187 forks source link

imagenet训练代码可以公布吗? #50

Open brisker opened 3 years ago

brisker commented 3 years ago

imagenet训练代码可以公布吗?

HantingChen commented 3 years ago

You can directly modify the Pytorch official imagenet training example: https://github.com/pytorch/examples/blob/master/imagenet/main.py, by replacing the conv-resnet with adder-resnet and using cosine learning rate decay with 150 epochs training.

brisker commented 3 years ago

@HantingChen 如果你们暂时没有cuda版本的addernet开源计划, 可否提供你们的resnet-imagenet和resnet-cifar的训练log以供参考呢?