Skip to content
This repository has been archived by the owner on Oct 31, 2023. It is now read-only.

CUDA out of memory #138

Open
aooating opened this issue Mar 1, 2023 · 0 comments
Open

CUDA out of memory #138

aooating opened this issue Mar 1, 2023 · 0 comments

Comments

@aooating
Copy link

aooating commented Mar 1, 2023

my codes are as follows:
with higher.innerloop_ctx(model, model_opt) as (fmodel, diffopt):
for j in range(param_config['inner_steps']):
buff_out = fmodel(buff_imgs)
buff_loss = criterion(.......)
diffopt.step(buff_loss)
After each inner loop, CUDA memory increases until the cache explodes.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant