Simple testing model for Kipoi/pytorch by Roman Kreuzhuber