search

HuiZeng / Grid-Anchor-based-Image-Cropping

Project page of the CVPR2019 paper "Reliable and Efficient Image Cropping: A Grid Anchor based Approach"
114 stars 12 forks source link

Can I change the epoches of the train？ #11

Closed WujiangXu closed 5 years ago

WujiangXu commented 5 years ago

I want to increase the epoches of the train or the times of the train in order to improve the training model of my dataset.Or are there any other ways to improve the training model.

HuiZeng commented 5 years ago

You can make any change in the matlab training code. For pytorch code, I currently did not release the training code. You can implement it by yourself and make any change as your like. I will release the pytorch training code together with a journal version which has several improvements.

WujiangXu commented 5 years ago

Thank u for u answer. I just use the code of matlab version.But I can't find the place to change to increase the times of the train or the epoches of the train.Whether it is set by the vgg16 or resnet50.

HuiZeng commented 5 years ago

This is a trivial implementation in the matconvnet. You can change the setting of learning rate in the model initialization function. The epoch will automatically adjust to the length of the learning rate. image

WujiangXu commented 5 years ago

thank you for your reply.I found it several minutes ago together.：）