fix: self.soft_argmax used but never instantiated

[andrewwhitecdw]

Problem

fix: self.soft_argmax used but never instantiated

Fix

Replace:

        # Optimizer is created in a separate call, because this isn't needed unless we're training
        self.optimizer = None

with:

        self.soft_argmax = SoftArgmaxPavlo(self.n_keypoints)

        # Optimizer is created in a separate call, because this isn't needed unless we're training
        self.optimizer = None

Files changed

• dream/network.py

[andrewwhitecdw]

Closing this sweep-generated PR: sole commit is missing a valid Signed-off-by trailer. It does not meet the sweep requirements (single signed-off commit).