added workdir parameter

2025-12-17 13:46:46 +01:00
parent 2c2625c778
commit 4d05a59d0b
5 changed files with 19 additions and 14 deletions
@@ -4,8 +4,8 @@ from cd_train import cd_jens
 from stack import Stack, StackType

 class StackDeep(Stack):
-	def __init__(self, name: str):
-		Stack.__init__(self, StackType.Deep, name)
+	def __init__(self, name: str, work_dir: str = '.'):
+		Stack.__init__(self, StackType.Deep, name, work_dir)

 	def batch_from(self, batch: np.ndarray, from_layer_id: int = 0):
 		_batch = np.matrix.copy(batch)