INNER CODE UNIT · Python

loss

QizhiPei/FABind · FABind/fabind/main_fabind.py:414

        loss = com_coord_loss + \
            contact_loss + contact_by_pred_loss + contact_distill_loss + \
            pocket_cls_loss + \
            pocket_coord_loss
        
        accelerator.backward(loss)
        if args.clip_grad:
            # clip_grad_norm_(model.parameters(), max_norm=1.0, error_if_nonfinite=True)
            if accelerator.sync_gradients:
                accelerator.clip_grad_norm_(model.parameters(), max_norm=1.0)
        
        optimizer.step()
        scheduler.step()

        batch_loss += len(y_pred)*contact_loss.item()
        batch_by_pred_loss += len(y_pred_by_coord)*contact_by_pred_loss.item()
        batch_distill_loss += len(y_pred_by_coord)*contact_distill_loss.item()
        com_coord_batch_loss += len(com_coord_pred)*com_coord_loss.item()

View source record →