• Davis King's avatar
    Upgraded to cuDNN v4.0. This means changing the binding to max_pool a little · 122f2fa6
    Davis King authored
    since that's a little different in cuDNN.  I also removed my CUDA code for
    doing batch normalization and replaced it with cuDNN's new batch normalization
    methods.
    
    Finally, I forgot to add a convolutional option to the bn_ object.  Now it has
    one so you can set the mode however you like, either BATCH_NORM_FC or
    BATCH_NORM_CONV.
    122f2fa6
CMakeLists.txt 24.1 KB