search

snuvclab / gala

Apache License 2.0
144 stars 12 forks source link

Out of memory #3

Closed fcyx closed 4 months ago

fcyx commented 4 months ago

Hello, your work is excellent, but I encountered an issue with insufficient memory when trying to reproduce your work. My hardware configuration is 4090 24GB. Specifically, I encounter out-of-memory errors during the training process of the Appearance Stage.

fcyx commented 4 months ago

According to what I've seen, the default batchsize = 1 in the code is already the minimum. I'm not sure what other adjustments should be made.

fcyx commented 4 months ago

Sorry, I made a mistake in reading the file. The batch size defaults to 5 and can be set to 1 to complete the run