Commit 33a525b9 authored by Dennis Noll's avatar Dennis Noll
Browse files

[recipes] common: remove HACK (now done in network with blackout)

parent e55cb4d6
......@@ -398,9 +398,6 @@ class DNNBase:
groups = self.tensors[k][-1].get("groups", [])
if "multiclass" in groups and "eval" in groups:
dnn_inputs[k] = v[mask]
dnn_inputs["hl"] = dnn_inputs["hl"][
:, :30
] # HACK: used for running PlotProducer to inspect variables before doing new training
(pm,) = self.dnn(inputs=dnn_inputs).values()
target_shape = (mask.sum(), len(self.multiclass_processes))
assert pm.shape == target_shape, (pm.shape, target_shape)
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment