Commit 3ecf4189 authored by Cresson Remi's avatar Cresson Remi
Browse files

FIX: dataset minig lock

Showing with 1 addition and 1 deletion
+1 -1
...@@ -394,12 +394,12 @@ class Dataset: ...@@ -394,12 +394,12 @@ class Dataset:
# buffers # buffers
self.miner_buffer = Buffer(buffer_length) self.miner_buffer = Buffer(buffer_length)
self.mining_lock = multiprocessing.Lock()
self.consumer_buffer = Buffer(buffer_length) self.consumer_buffer = Buffer(buffer_length)
self.consumer_buffer_pos = 0 self.consumer_buffer_pos = 0
self.tot_wait = 0 self.tot_wait = 0
self.miner_thread = self._summon_miner_thread() self.miner_thread = self._summon_miner_thread()
self.read_lock = multiprocessing.Lock() self.read_lock = multiprocessing.Lock()
self.mining_lock = multiprocessing.Lock()
self._dump() self._dump()
# Prepare tf dataset for one epoch # Prepare tf dataset for one epoch
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment