facebookresearch / VMZ

VMZ: Model Zoo for Video Modeling
Apache License 2.0
1.04k stars 155 forks source link

caffe equivalent of iter_size in caffe2? #21

Closed liu-zhy closed 6 years ago

liu-zhy commented 6 years ago

Iter_size in caffe solver enables to perform backpropagation after a number of forward passes defined by the value of iter_size. Hence, allowing an increased batch size without GPU memory restriction. Is there an equivalent in caffe2 that can be used with optimizer.build_sgd?