Just change max number of epochs to small value for quick cluster testing.