accuracy - Githubissues

USTC-Video-Understanding / I3D_Finetune

TensorFlow code for finetuning I3D model on UCF101.

144 stars 43 forks source link

accuracy #12

Open panna19951227 opened 5 years ago

panna19951227 commented 5 years ago

2018-09-13 04:56:39.376762: W tensorflow/core/common_runtime/bfc_allocator.cc:219] Allocator (GPU_0_bfc) ran out of memory trying to allocate 2.61GiB. The caller indicates that this is not a failure, but may mean that there could be performance gains if more memory were available.

volatile GPU-Util is very low,where can I adjust in the codes to reduce memory.

AlexHu123 commented 5 years ago

hi panna, you can reduce the batchsize(BATCH_SIZE) or the frame rate(_CLIP_SIZE ) in finetune.py.