Jump to content
Compvision.ru

Recommended Posts

11 minutes ago, mrgloom said:

ну меня всё равно порадовало, что на моем железе GPU даёт прирост в ~x15.

да и на хабре использовали ноутбучную видеокарту NVIDIA GeForce GT 540M.

Сколько же занимает прогон MNIST на современных видеокартах, может кто то попробовать? (можно даже и не под виндой, а взять protobuf и данные из вышеобозначенного архива)

 

Попробую собрать полноценно через https://github.com/willyd/caffe-builder

Что вы подразумеваете под полным прогоном? До точности классификации 99.8% на моей GTX980Ti обучается ~10 секунд.

p.s. в своем эксперименте исопльзовал топологию сети LeNet

Share this post


Link to post
Share on other sites

Я подразумеваю те настройки которые в protobuf.

https://drive.google.com/file/d/0B88eZEyituf8YUlROE5oZWpXSkE/

Они вроде как отличаются немного от тех что лежат у стандартного caffe

https://github.com/BVLC/caffe/tree/master/examples/mnist

 

Share this post


Link to post
Share on other sites
30 minutes ago, mrgloom said:

Я подразумеваю те настройки которые в protobuf.

https://drive.google.com/file/d/0B88eZEyituf8YUlROE5oZWpXSkE/

Они вроде как отличаются немного от тех что лежат у стандартного caffe

https://github.com/BVLC/caffe/tree/master/examples/mnist

 

Все итерации что ли пробежать? Вечером попробую. Но не думаю, что время работы будет сильно отличаться, т.к. в MNIST картинки маленькие, и современным картам разгуляться негде. Спидапы лучше при тренировке на ImageNet'е каком-нибудь тестировать :)

Share this post


Link to post
Share on other sites

MNIST 28x28 grayscale, LeNet, 30 эпох, 1 минута 25 секунд на GTX 970.

Со стандартным размером бэтча, если увеличить то до 45 сек.

Тестил на DIGITS 3.

Share this post


Link to post
Share on other sites
Цитата

30 эпох

это что значит эпох? там вроде в терминах итераций задается.

типа

# The maximum number of iterations
max_iter: 20000

Share this post


Link to post
Share on other sites

21120 итераций.

Эпоха, это один полный проход по всем сэмплам, количество итераций еще зависит от размера бэтча.

Share this post


Link to post
Share on other sites

Эпохи задаются только через DIGITS? или это вы примерно посчитали n_iter*batch_size/train_size

Share this post


Link to post
Share on other sites
1 hour ago, Smorodov said:

21120 итераций.

Эпоха, это один полный проход по всем сэмплам, количество итераций еще зависит от размера бэтча.

А как он узнает, что прошелся по всем самплам? В caffe ведь, вроде бы batch каждый раз из базы берется случайным образом...

Share this post


Link to post
Share on other sites

Насколько я понимаю, берет он по-порядку, а вот при создании базы перемешивает, если задан такой флаг.

Share this post


Link to post
Share on other sites

Create an account or sign in to comment

You need to be a member in order to leave a comment

Create an account

Sign up for a new account in our community. It's easy!

Register a new account

Sign in

Already have an account? Sign in here.

Sign In Now


  • Recently Browsing   0 members

    No registered users viewing this page.

×