search

PatrykChrabaszcz / Imagenet32_Scripts

Scripts for Imagenet 32 dataset
MIT License
150 stars 47 forks source link

Batch size #4

Closed tangbohu closed 5 years ago

tangbohu commented 6 years ago

Hi, What batch size did you set on these experiments? Thanks

tangbohu commented 6 years ago

BTW, did you use SGD on these experiments reported in the paper.

littlesnowperfume commented 5 years ago

I also have this question about what is the batch_size number??

littlesnowperfume commented 5 years ago

HELLO: @tangbohu could we change e-mails to each others?? my email is paul8301526@gmail.com Maybe we could discuss this paper?? My wechat ID: paul8301526 HELLO: I found he use batchsize=128 when use the GPU(Titan X) image

PatrykChrabaszcz commented 5 years ago

Hi, Indeed, it's 128: https://github.com/PatrykChrabaszcz/Imagenet32_Scripts/blob/master/WRNs_imagenet.py#L318

It's SGD with momentum: https://github.com/PatrykChrabaszcz/Imagenet32_Scripts/blob/master/WRNs_imagenet.py#L297