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 { |
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 = ( |