Faster batch normalization

This commit is contained in:
Joseph Redmon
2015-12-07 17:18:04 -08:00
parent b4d81aa7bd
commit c2738835f0
7 changed files with 149 additions and 100 deletions

View File

@ -20,7 +20,8 @@ image coco_labels[80];
void train_coco(char *cfgfile, char *weightfile)
{
//char *train_images = "/home/pjreddie/data/voc/test/train.txt";
char *train_images = "/home/pjreddie/data/coco/train.txt";
//char *train_images = "/home/pjreddie/data/coco/train.txt";
char *train_images = "data/coco.trainval.txt";
char *backup_directory = "/home/pjreddie/backup/";
srand(time(0));
data_seed = time(0);