Use some traits
Base support for conv rbm training