You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Hi, I was reproducing TPT with loading pretrained weight from COOP .
I realized the current code directly loads pretrained weight which is mapped at gpu index:0.
This causes the current code to map at global gpu index 0 which is not what I want.
I think the loading pretrained context should be mapped to 'cpu' and then copy the weight from there.
This will save some memory that was holding up pretrained weight.
The text was updated successfully, but these errors were encountered:
SeonghaEom
changed the title
CUDA device error when using with pretrained weight from COOP
CUDA out of memory issue when using with pretrained weight from COOP
Jun 13, 2023
Hi, I was reproducing TPT with loading pretrained weight from COOP .
I realized the current code directly loads pretrained weight which is mapped at gpu index:0.
This causes the current code to map at global gpu index 0 which is not what I want.
I think the loading pretrained context should be mapped to 'cpu' and then copy the weight from there.
This will save some memory that was holding up pretrained weight.
The text was updated successfully, but these errors were encountered: