[
https://issues.apache.org/jira/browse/SINGA-55?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel
]
wangwei resolved SINGA-55.
--------------------------
Resolution: Fixed
Assignee: wangwei
> Refactor main.cc and singa.h
> ----------------------------
>
> Key: SINGA-55
> URL: https://issues.apache.org/jira/browse/SINGA-55
> Project: Singa
> Issue Type: Improvement
> Reporter: wangwei
> Assignee: wangwei
>
> This ticket is a continuation of SINGA-36.
> A Driver class will be provided, which includes the following functions:
> * Init() to init glogs, gflags (may remove the dependency on flags later);
> register built-in layers, updaters, etc.; setup folders; init MPI if MPI is
> used for message passing.
> * RegisterLayer/Updater/Worker/Param, for registering user defined layers,
> updaters, etc.
> * Submit() for submitting the job configuration.
> The user's main program would then look like :
> {code}
> #include "singa.h"
> int main(int argc, char** argv) {
> singa::Driver driver;
> dirver.Init(argc, argv);
> dirver.RegisterLayer<SubLayerClass>(sublayerID);
> dirver.RegisterUpdater<SubUpdaterClass>(subUpdaterID);
> ...
> // prepare job conf
> driver.Submit(..);
> return 0;
> }
> {code}
--
This message was sent by Atlassian JIRA
(v6.3.4#6332)