minor fix

This commit is contained in:
AlexeyAB 2020-11-18 04:06:28 +03:00
parent 17cf4bfbfc
commit 5c046bea96
3 changed files with 1430 additions and 1 deletions

View File

@ -1131,6 +1131,7 @@ size=1
stride=1
pad=1
activation=mish
stopbackward=800
##########################

1428
cfg/yolov4x-mish.cfg Normal file

File diff suppressed because it is too large Load Diff

View File

@ -947,7 +947,7 @@ void forward_yolo_layer_gpu(const layer l, network_state state)
// if(y->1) x -> inf
// if(y->0) x -> -inf
if (l.new_coords) {
activate_array_ongpu(l.output_gpu + index, 4 * l.w*l.h, LOGISTIC); // x,y
activate_array_ongpu(l.output_gpu + index, 4 * l.w*l.h, LOGISTIC); // x,y,w,h
}
else {
activate_array_ongpu(l.output_gpu + index, 2 * l.w*l.h, LOGISTIC); // x,y