singa-dev mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From "wangwei (JIRA)" <j...@apache.org>
Subject [jira] [Created] (SINGA-328) Add VGG models for ImageNet classification
Date Tue, 04 Jul 2017 06:10:00 GMT
wangwei created SINGA-328:
-----------------------------

             Summary: Add VGG models for ImageNet classification
                 Key: SINGA-328
                 URL: https://issues.apache.org/jira/browse/SINGA-328
             Project: Singa
          Issue Type: New Feature
            Reporter: wangwei
            Assignee: wangwei


There are 4 VGG models proposed in the paper https://arxiv.org/abs/1409.1556.
These 4 models are later updated by adding batch-normalization layers.
In this ticket, we convert the pre-trained parameters from [pytorch|https://github.com/pytorch/vision/blob/master/torchvision/models/vgg.py]
into SINGA. Users can then fine-tune the VGG models for their own data using the pre-trained
parameters.



--
This message was sent by Atlassian JIRA
(v6.4.14#64029)

Mime
View raw message