Skip to content

Instantly share code, notes, and snippets.

@Dipeshpal
Created October 14, 2020 14:35
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save Dipeshpal/68ed40b1c9bbacc5e51aff0831f760b0 to your computer and use it in GitHub Desktop.
Save Dipeshpal/68ed40b1c9bbacc5e51aff0831f760b0 to your computer and use it in GitHub Desktop.
Found 200 files belonging to 2 classes.
Using 160 files for training.
2020-10-14 20:03:49.722610: W tensorflow/stream_executor/platform/default/dso_loader.cc:59] Could not load dynamic library 'nvcuda.dll'; dlerror: nvcuda.dll not found
2020-10-14 20:03:49.722866: W tensorflow/stream_executor/cuda/cuda_driver.cc:312] failed call to cuInit: UNKNOWN ERROR (303)
2020-10-14 20:03:49.729797: I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:169] retrieving CUDA diagnostic information for host: Adonis-PC
2020-10-14 20:03:49.730065: I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:176] hostname: Adonis-PC
2020-10-14 20:03:49.753847: I tensorflow/core/platform/cpu_feature_guard.cc:142] This TensorFlow binary is optimized with oneAPI Deep Neural Network Library (oneDNN)to use the following CPU instructions in performance-critical operations: AVX2
To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.
2020-10-14 20:03:50.029289: I tensorflow/compiler/xla/service/service.cc:168] XLA service 0x25ee96ca170 initialized for platform Host (this does not guarantee that XLA will be used). Devices:
2020-10-14 20:03:50.029612: I tensorflow/compiler/xla/service/service.cc:176] StreamExecutor device (0): Host, Default Version
Found 200 files belonging to 2 classes.
Using 40 files for validation.
Dataset Dir: datasets
Number of class: 2
Total Images: 200
batch_size: 32
img_width: 128
img_height: 128
class_names: ['Dipesh', 'Jay']
Model Directory: model/model.h5
================================================
================= Starting Training =================
Model: "sequential_1"
_________________________________________________________________
Layer (type) Output Shape Param #
=================================================================
sequential (Sequential) (None, 128, 128, 3) 0
_________________________________________________________________
rescaling (Rescaling) (None, 128, 128, 3) 0
_________________________________________________________________
conv2d (Conv2D) (None, 128, 128, 16) 448
_________________________________________________________________
max_pooling2d (MaxPooling2D) (None, 64, 64, 16) 0
_________________________________________________________________
conv2d_1 (Conv2D) (None, 64, 64, 32) 4640
_________________________________________________________________
max_pooling2d_1 (MaxPooling2 (None, 32, 32, 32) 0
_________________________________________________________________
conv2d_2 (Conv2D) (None, 32, 32, 64) 18496
_________________________________________________________________
max_pooling2d_2 (MaxPooling2 (None, 16, 16, 64) 0
_________________________________________________________________
flatten (Flatten) (None, 16384) 0
_________________________________________________________________
dense (Dense) (None, 128) 2097280
_________________________________________________________________
dense_1 (Dense) (None, 2) 258
=================================================================
Total params: 2,121,122
Trainable params: 2,121,122
Non-trainable params: 0
_________________________________________________________________
Epoch 1/10
5/5 [==============================] - 2s 399ms/step - loss: 0.7228 - accuracy: 0.4750 - val_loss: 0.5305 - val_accuracy: 0.5500
Epoch 2/10
5/5 [==============================] - 2s 309ms/step - loss: 0.4483 - accuracy: 0.8750 - val_loss: 0.3366 - val_accuracy: 0.9250
Epoch 3/10
5/5 [==============================] - 2s 318ms/step - loss: 0.3335 - accuracy: 0.8875 - val_loss: 0.2336 - val_accuracy: 0.9500
Epoch 4/10
5/5 [==============================] - 2s 321ms/step - loss: 0.1870 - accuracy: 0.9625 - val_loss: 0.1201 - val_accuracy: 0.9500
Epoch 5/10
5/5 [==============================] - 2s 366ms/step - loss: 0.0699 - accuracy: 0.9812 - val_loss: 0.1016 - val_accuracy: 0.9500
Epoch 6/10
5/5 [==============================] - 2s 350ms/step - loss: 0.0594 - accuracy: 0.9750 - val_loss: 0.1444 - val_accuracy: 0.9500
Epoch 7/10
5/5 [==============================] - 2s 396ms/step - loss: 0.0647 - accuracy: 0.9812 - val_loss: 0.0892 - val_accuracy: 0.9500
Epoch 8/10
5/5 [==============================] - 2s 318ms/step - loss: 0.0301 - accuracy: 0.9812 - val_loss: 0.1556 - val_accuracy: 0.9500
Epoch 9/10
5/5 [==============================] - 2s 328ms/step - loss: 0.0543 - accuracy: 0.9875 - val_loss: 0.1214 - val_accuracy: 0.9500
Epoch 10/10
5/5 [==============================] - 2s 322ms/step - loss: 0.0234 - accuracy: 0.9937 - val_loss: 0.0225 - val_accuracy: 1.0000
Process finished with exit code 0
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment