mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

stable, dropout on gpu

Browse Source
This commit is contained in:
Joseph Redmon 2014-11-19 14:03:51 -08:00
parent d407bffde9
commit 7c120aef23
2 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/cnn.c
View File

@ -308,8 +308,8 @@ void train_asirra()
void train_imagenet() void train_imagenet()
{ {
float avg_loss = 1; float avg_loss = 1;
network net = parse_network_cfg("/home/pjreddie/imagenet_backup/imagenet_2280.cfg"); //network net = parse_network_cfg("/home/pjreddie/imagenet_backup/alexnet_1270.cfg");
//network net = parse_network_cfg("cfg/imagenet2.cfg"); network net = parse_network_cfg("cfg/imagenet.cfg");
printf("Learning Rate: %g, Momentum: %g, Decay: %g\n", net.learning_rate, net.momentum, net.decay); printf("Learning Rate: %g, Momentum: %g, Decay: %g\n", net.learning_rate, net.momentum, net.decay);
int imgs = 1000/net.batch+1; int imgs = 1000/net.batch+1;
srand(time(0)); srand(time(0));
@ -1042,6 +1042,7 @@ int main(int argc, char *argv[])
#ifdef GPU #ifdef GPU
else if(0==strcmp(argv[1], "test_gpu")) test_gpu_blas(); else if(0==strcmp(argv[1], "test_gpu")) test_gpu_blas();
#endif #endif
test_parser();
fprintf(stderr, "Success!\n"); fprintf(stderr, "Success!\n");
return 0; return 0;
} }

5
src/dropout_layer.cl Normal file
View File

@ -0,0 +1,5 @@
__kernel void forward(__global float *input, __global float *rand, float prob)
{
int id = get_global_id(0);
input[id] = (rand[id] < prob) ? 0 : input[id]/(1.-prob);
}