Daquan Lin
Daquan Lin
REMOVE ":" in your environment files, such as ~/.bashrc or ~/.zshrc says, "export CUDA_HOME=$CUDA_HOME:/usr/local/cuda" change to "export CUDA_HOME=/usr/local/cuda"
Hi @patil-suraj , I tried to load `unet_with_ema` in `EMAModel` and `unet_without_ema` for fine-tuning, but the result remains terrible. Could you help me fix it? Thanks. Here are my code...
Cool thanks. In my experiment, I got the same results, the setting was as below: 4*NVIDIA-V100-PCIE-16GB, batch_size=2, gradient_accumulation_steps=3, mixed_precision="fp16" deepspeed_stage=2
Hi @wangleiofficial, I met the problem same with you. Do you fix it?