Stuck at some place 2

[Anusingh5012]

I try to run train.py (python train.py --prepare_data True --dataset_dir dataset --log_dir logs --model_version 2 --image_height 321 --image_width 321 --train_batch_size 40 --test_batch_size 200 --max_iteration 10000), but the code stuck at "2018-05-09 11:16:34.072660: I T:\src\github\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:1053] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 9606 MB memory) -> physical GPU (device: 0, name: TITAN Xp, pci bus id: 0000:03:00.0, compute capability: 6.1)".

My code doesn't terminate, it just hangs there and stop giving any output.

Could you please help me to solve the problem?

Im using GPU - NVIDIA TITAN Xp Tensorflow- 1.8.0

[clausmichele]

I solved commenting the testing part.

[Anusingh5012]

Exactly!!!...Now I am able to run. Thank You.

On Wed, Jun 13, 2018 at 2:38 PM, clausmichele notifications@github.com wrote:

The problem was prepare_data.py, it was taking only 5 folders and the test.tfrecords was empy.

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub https://github.com/SaoYan/bgsCNN/issues/7#issuecomment-396869394, or mute the thread https://github.com/notifications/unsubscribe-auth/AEAADlcKgfYwGqrZxI1nt94egB0WFdTTks5t8NaqgaJpZM4T3vsx .

--

Thanks & Regards

Mr. Anu Singha MCA, M.Tech(CSE), PhD (pursuing) South Asian University, Tripura University Suryamaninagar-799022, Agartala, INDIA Contact No: 8974310941 Email-Id:anusingh5012@gmail.com Website: https://sites.google.com/site/anusingh01031989/