diff --git a/src/main/java/qupath/ext/biop/cellpose/Cellpose2D.java b/src/main/java/qupath/ext/biop/cellpose/Cellpose2D.java
index ffbe6d3..c7a2b8d 100644
--- a/src/main/java/qupath/ext/biop/cellpose/Cellpose2D.java
+++ b/src/main/java/qupath/ext/biop/cellpose/Cellpose2D.java
@@ -671,7 +671,8 @@ private void runCellposeTraining() throws IOException, InterruptedException {
 
         if (useGPU) cellposeArguments.add("--use_gpu");
 
-        if (cellposeSetup.getVersion().equals(CellposeSetup.CellposeVersion.CELLPOSE_1))
+        if ( cellposeSetup.getVersion().equals(CellposeSetup.CellposeVersion.CELLPOSE_1) ||
+                cellposeSetup.getVersion().equals(CellposeSetup.CellposeVersion.CELLPOSE_2) )
             cellposeArguments.add("--verbose");
 
         veRunner.setArguments(cellposeArguments);