set_mode to CPU in ArgMaxLayerTest constructor to avoid random Travis
failures.