Joseph Redmon
2014-12-08 a99050f0c8cb0315fa31e3d1fa3e38594fe5e40a
src/cnn.c
@@ -374,7 +374,7 @@
void train_imagenet_distributed(char *address)
{
    float avg_loss = 1;
    srand(0);
    srand(time(0));
    network net = parse_network_cfg("cfg/alexnet.client");
    printf("Learning Rate: %g, Momentum: %g, Decay: %g\n", net.learning_rate, net.momentum, net.decay);
    int imgs = 1000/net.batch+1;
@@ -412,11 +412,11 @@
{
    float avg_loss = 1;
    //network net = parse_network_cfg("/home/pjreddie/imagenet_backup/alexnet_1270.cfg");
    srand(0);
    srand(time(0));
    network net = parse_network_cfg("cfg/alexnet.cfg");
    printf("Learning Rate: %g, Momentum: %g, Decay: %g\n", net.learning_rate, net.momentum, net.decay);
    int imgs = 1000/net.batch+1;
    imgs=1;
    //imgs=1;
    int i = 0;
    char **labels = get_labels("/home/pjreddie/data/imagenet/cls.labels.list");
    list *plist = get_paths("/data/imagenet/cls.train.list");
@@ -872,7 +872,7 @@
void run_server()
{
    srand(0);
    srand(time(0));
    network net = parse_network_cfg("cfg/alexnet.server");
    server_update(net);
}