Le Thi Thanh Van

os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = 'true'

@yjxiong when I test model with RGB, it appears error as below: RuntimeError: Error(s) in loading state_dict for BNInception: Please give your pytorch version that you used to train