Stable, needs to be way faster

This commit is contained in:
Joseph Redmon
2015-01-14 12:18:57 -08:00
parent aa5996d58e
commit 08b757a0bf
4 changed files with 37 additions and 7 deletions

View File

@ -15,6 +15,35 @@
#include "softmax_layer.h"
#include "dropout_layer.h"
char *get_layer_string(LAYER_TYPE a)
{
switch(a){
case CONVOLUTIONAL:
return "convolutional";
case CONNECTED:
return "connected";
case MAXPOOL:
return "maxpool";
case SOFTMAX:
return "softmax";
case NORMALIZATION:
return "normalization";
case DROPOUT:
return "dropout";
case FREEWEIGHT:
return "freeweight";
case CROP:
return "crop";
case COST:
return "cost";
default:
break;
}
return "none";
}
network make_network(int n, int batch)
{
network net;