LambdaLabsML / examples

Deep Learning Examples
MIT License
805 stars 103 forks source link

TypeError: Gradient accumulation supports only int and dict types #46

Open GuangguangLu opened 1 year ago

GuangguangLu commented 1 year ago

TypeError: Gradient accumulation supports only int and dict types. How to deal with this problem?

offchan42 commented 1 year ago

Are you using the pokemon fine tune example? Fix the typo in the notebook. There's a reference to a variable called NUM_GPUS but it doesn't exist. The actual name is N_GPUS.

Also follow this advice: https://github.com/justinpinkney/stable-diffusion/issues/24#issuecomment-1312419797