Update autoPyTorch/pipeline/components/setup/network_backbone/utils.py

ArlindKadra · nabenabe0928 · web-flow · commit 14889783d669 · 2021-10-21T18:13:30.000+02:00
Providing information on the wrong method provided for shake-shake regularization.

Co-authored-by: nabenabe0928 &lt;47781922+nabenabe0928@users.noreply.github.com&gt;
diff --git a/autoPyTorch/pipeline/components/setup/network_backbone/utils.py b/autoPyTorch/pipeline/components/setup/network_backbone/utils.py
@@ -140,7 +140,7 @@ def shake_get_alpha_beta(
             [rnd * (0.5 - alpha) + alpha if alpha < 0.5 else rnd * (alpha - 0.5) + 0.5]
         )
     else:
-        raise ValueError("Unknown method for ShakeShakeRegularisation in NetworkBackbone")
+        raise ValueError(f"Unknown method `{method}` for ShakeShakeRegularisation in NetworkBackbone")
 
     if is_cuda:
         alpha = alpha.cuda()

Original file line number	Diff line number	Diff line change
`@@ -140,7 +140,7 @@ def shake_get_alpha_beta(`
`140`	`140`	`[rnd * (0.5 - alpha) + alpha if alpha < 0.5 else rnd * (alpha - 0.5) + 0.5]`
`141`	`141`	`)`
`142`	`142`	`else:`
`143`		`- raise ValueError("Unknown method for ShakeShakeRegularisation in NetworkBackbone")`
	`143`	+ raise ValueError(f"Unknown method `{method}` for ShakeShakeRegularisation in NetworkBackbone")
`144`	`144`
`145`	`145`	`if is_cuda:`
`146`	`146`	`alpha = alpha.cuda()`