issues
search
google-research
/
big_transfer
Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.
https://arxiv.org/abs/1912.11370
Apache License 2.0
1.5k
stars
175
forks
source link
fix issue on geforce GPU
#15
Open
prob1995
opened
4 years ago
prob1995
commented
4 years ago
Enable memory_growth to prevent OOM.
Fix variable to prevent TypeError (may multiply a NoneType).