Ben Jami

different loss from GPU

1

I trained two ALS models for the same data on GPU and CPU, but for GPU, I have 0.06 for the loss, and in CPU, I have 0.0006 for loss....

Hello, I've encountered an unexpected behavior when using torchdrug alongside PyTorch. Specifically, after importing torchdrug, the memory address of torch.nn.Module changes, which leads to issues when checking instances of nn.Module....

Difficulty Replicating Antibody Optimization Results

2

I've been trying to replicate the antibody optimization results detailed in the paper, strictly adhering to the provided instructions and ensuring the activation of OpenMM relaxation. However, my attempts to...

GKD trainer + chunked JSD loss + FSDP

1

Hello Liger Kernel team, First of all, thank you for making this project available! I’ve been exploring your codebase and tried to implement GKDTrainer using the chunked_jsd_loss similarly to how...