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
  1. Enable memory_growth to prevent OOM.
  2. Fix variable to prevent TypeError (may multiply a NoneType).