Skip to content

fix: load frozen autoencoder checkpoint on CPU to avoid DDP hangs#10

Open
DannyP0 wants to merge 1 commit into
NVIDIA-BioNeMo:mainfrom
DannyP0:main
Open

fix: load frozen autoencoder checkpoint on CPU to avoid DDP hangs#10
DannyP0 wants to merge 1 commit into
NVIDIA-BioNeMo:mainfrom
DannyP0:main

fix: load VAE checkpoint with map_location=cpu for multi-GPU training

b3c8c91
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs