Jackoatmon commited on
Commit
7de795d
·
verified ·
1 Parent(s): 73bb0f5

Update Feather training runtime image

Browse files
Files changed (1) hide show
  1. overlay/hydra/model.py +1 -1
overlay/hydra/model.py CHANGED
@@ -508,7 +508,7 @@ class PostSemClawModel(nn.Module):
508
 
509
  total_assigned = len(embedding_params) + len(lm_head_params) + len(matrix_params) + len(scalar_params)
510
  total_params = len(list(self.parameters()))
511
- sdr_excluded = len(list(self.sdr_semantic.parameters()))
512
  assert total_assigned + sdr_excluded == total_params, (
513
  f"Parameter count mismatch: assigned {total_assigned} + sdr_excluded "
514
  f"{sdr_excluded} vs total {total_params}"
 
508
 
509
  total_assigned = len(embedding_params) + len(lm_head_params) + len(matrix_params) + len(scalar_params)
510
  total_params = len(list(self.parameters()))
511
+ sdr_excluded = len(list(self.sdr_semantic.parameters())) if self.sdr_semantic is not None else 0
512
  assert total_assigned + sdr_excluded == total_params, (
513
  f"Parameter count mismatch: assigned {total_assigned} + sdr_excluded "
514
  f"{sdr_excluded} vs total {total_params}"