Closed noelcodes closed 5 years ago
I ran training, but hit this error... Anyone getting this?
python train.py --dataset UCSD --dataset_address ./dataset/UCSD_Anomaly_Dataset.v1p2/UCSDped2/Train --input_height 45 --output_height 45 {'attention_label': 1, 'batch_size': 128, 'beta1': 0.5, 'checkpoint_dir': 'checkpoint', 'dataset': 'UCSD', 'dataset_address': './dataset/UCSD_Anomaly_Dataset.v1p2/UCSDped2/Train', 'epoch': 40, 'input_fname_pattern': '*', 'input_height': 45, 'input_width': None, 'learning_rate': 0.002, 'log_dir': 'log', 'output_height': 45, 'output_width': None, 'r_alpha': 0.2, 'sample_dir': 'samples', 'train': True, 'train_size': inf} 2018-12-22 15:03:46.625140: I C:\tf_jenkins\home\workspace\rel-win\M\windows-gpu\PY\35\tensorflow\core\platform\cpu_feature_guard.cc:137] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX AVX2 2018-12-22 15:03:46.948538: I C:\tf_jenkins\home\workspace\rel-win\M\windows-gpu\PY\35\tensorflow\core\common_runtime\gpu\gpu_device.cc:1030] Found device 0 with properties: name: GeForce GTX 1060 major: 6 minor: 1 memoryClockRate(GHz): 1.6705 pciBusID: 0000:01:00.0 totalMemory: 6.00GiB freeMemory: 4.97GiB 2018-12-22 15:03:46.965900: I C:\tf_jenkins\home\workspace\rel-win\M\windows-gpu\PY\35\tensorflow\core\common_runtime\gpu\gpu_device.cc:1120] Creating TensorFlow device (/device:GPU:0) -> (device: 0, name: GeForce GTX 1060, pci bus id: 0000:01:00.0, compute capability: 6.1) Program is on Train Mode libpng error: Write Error
Ah.. its working now after updating:
conda update -c c3i_test2 libpng
I ran training, but hit this error... Anyone getting this?