Sergio Guadarrama sguada
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| def make_reverb_dataset_double_interleave( | |
| trainer: str, | |
| num_parallel_calls: int = 8, | |
| prefetch: int = 0, | |
| batch_size: int = 2048, | |
| max_in_flight_samples_per_worker = 512, | |
| obs_dim: int = 32, | |
| action_dim: int = 2) -> tf.data.Dataset: | |
| all_shapes = ( |
sguada
/ blocking_queue.hpp
Created
December 24, 2014 23:26
BlockingQueue with maximum capacity blocks on Push (if full) and on Pop (if empty)
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #ifndef CAFFE_UTIL_BLOCKING_QUEUE_H_ | |
| #define CAFFE_UTIL_BLOCKING_QUEUE_H_ | |
| #include <queue> | |
| #include "boost/thread.hpp" | |
| namespace caffe { | |
| template<typename T> | |
| class BlockingQueue { |