fix: _freeze_stages uses undefined self.network instead of self.levels

[andrewwhitecdw]

Problem

fix: _freeze_stages uses undefined self.network instead of self.levels

Fix

Replace:

        if self.frozen_stages >= 2:
            for i in range(0, self.frozen_stages - 1):
                m = self.network[i]
                m.eval()
                for param in m.parameters():
                    param.requires_grad = False

with:

        if self.frozen_stages >= 2:
            for i in range(0, self.frozen_stages - 1):
                m = self.levels[i]
                m.eval()
                for param in m.parameters():
                    param.requires_grad = False

Files changed

• detection/models/gc_vit.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).