accidentally broke multigpu a while ago :-/

This commit is contained in:
Joseph Redmon
2017-06-26 14:00:18 -07:00
parent 63ebd694d7
commit 59ed1719d4
12 changed files with 119 additions and 46 deletions

View File

@ -23,6 +23,7 @@ void train_detector(char *datacfg, char *cfgfile, char *weightfile, int *gpus, i
cuda_set_device(gpus[i]);
#endif
nets[i] = load_network(cfgfile, weightfile, clear);
nets[i].learning_rate *= ngpus;
}
srand(time(0));
network net = nets[0];