Spaces:

HuangLab
/

CELL-E_2-Sequence_Prediction

Sleeping

Emaad commited on May 17, 2023

Commit

b42f002

•

1 Parent(s): 8580f27

Update celle_main.py

Files changed (1) hide show

celle_main.py CHANGED Viewed

@@ -198,7 +198,7 @@ class CELLE_trainer(pl.LightningModule):
  if k.startswith(ik):
  # print("Deleting key {} from state_dict.".format(k))
  del ckpt[k]
- self.load_state_dict(ckpt, strict=True)
  print(f"Restored from {path}")
  def forward(self, text, condition, target, return_loss=True):

  if k.startswith(ik):
  # print("Deleting key {} from state_dict.".format(k))
  del ckpt[k]
+ self.load_state_dict(ckpt, strict=False)
  print(f"Restored from {path}")
  def forward(self, text, condition, target, return_loss=True):