I believe this is correct dropout impl, ready to impl image-net