I tried for few hours, due to slow training I only trained for 1 epoch, this is a log
2021-07-26 23:09:28.130352: I tensorflow/core/grappler/optimizers/custom_graph_optimizer_registry.cc:112] Plugin optimizer for device_type GPU is enabled.
2021-07-26 23:09:28.185390: I tensorflow/core/grappler/optimizers/custom_graph_optimizer_registry.cc:112] Plugin optimizer for device_type GPU is enabled.
2021-07-26 23:09:28.217406: I tensorflow/core/grappler/optimizers/custom_graph_optimizer_registry.cc:112] Plugin optimizer for device_type GPU is enabled.
2021-07-26 23:09:28.229984: I tensorflow/core/grappler/optimizers/custom_graph_optimizer_registry.cc:112] Plugin optimizer for device_type GPU is enabled.
Epoch 1/1
20000/20000 [==============================] - loss: 0.5489 - accuracy: 0.6923
--- 6894.8485770225524902 seconds ---
Just for one epoch, it takes around 2 hours that's a nightmare
Topic:
Machine Learning & AI
SubTopic:
General
Tags: