Just a thought, try using Learner and not cnn_learner. Cnn_learner expects a split due to creating a custom head