Loss-graph store automatically (iterations == max_batches) at the end of tr...

AlexeyAB

2018-04-16 c52fa4742866125e35e88d0653dac99473e5fffb

 src/layer.h

@@ -281,7 +281,7 @@
    #ifdef CUDNN
    cudnnTensorDescriptor_t srcTensorDesc, dstTensorDesc;
    cudnnTensorDescriptor_t dsrcTensorDesc, ddstTensorDesc;
   cudnnTensorDescriptor_t normTensorDesc;
   cudnnTensorDescriptor_t normTensorDesc, normDstTensorDesc;
    cudnnFilterDescriptor_t weightDesc;
    cudnnFilterDescriptor_t dweightDesc;
    cudnnConvolutionDescriptor_t convDesc;