Method cache_to_cuda

roach/models/ppo_buffer.py:247–253

(self, batch_size)

        thread1.start()

    def cache_to_cuda(self, batch_size):
        self.sample_queue.queue.clear()

        for rollout_data in self.get(batch_size):
            while self.sample_queue.qsize() >= 2:
                time.sleep(0.01)
            self.sample_queue.put(rollout_data)

    def size(self) -> int:
        """