Closed philipkd closed 9 months ago
Steps to Reproduce:
train_weak_to_strong.py
Notice: "UnboundLocalError: local variable 'minibatch_size' referenced before assignment"
Colab notebook
[1]: It's not necessary to clone @fffffarmer's hotfix, but that's the only way I could get this repo working on Colab. You can probably reproduce this issue by simply running it on a single GPU.
Steps to Reproduce:
train_weak_to_strong.py
on a single GPUNotice: "UnboundLocalError: local variable 'minibatch_size' referenced before assignment"
Colab notebook
[1]: It's not necessary to clone @fffffarmer's hotfix, but that's the only way I could get this repo working on Colab. You can probably reproduce this issue by simply running it on a single GPU.