fix arguments to training network

This commit is contained in:
Dr. Kashif Rasul
2020-01-13 11:51:50 +01:00
parent c3744da5d7
commit 2d14e9d842
-1
View File
@@ -160,7 +160,6 @@ class TempFlowEstimator(PTSEstimator):
return TempFlowTrainingNetwork(
input_size=self.input_size,
target_dim=self.target_dim,
num_parallel_samples=self.num_parallel_samples,
num_layers=self.num_layers,
num_cells=self.num_cells,
cell_type=self.cell_type,