728x90 구축1 Tensorflow data pipeline 구축 Tensorflow에서 feed_dict로 데이터를 계속해서 공급하는 코드를 많이 보셨을것입니다. 그러나, 논문을 리뷰하면서 github 참조를 많이 하셨던 분들은 아시겠지만 feed_dict로 데이터를 공급하는 코드는 거의 없던것 같습니다. 또한 데이터를 로드하는 bottleneck 시간이 줄어들어서 학습시간이 줄어드는 효과가 있습니다. 이에 본 블로그에서는 data를 gpu에 계속해서 공급하는 tensorflow기반 dataset api를 참고해서 설명을 드리고자합니다. 1. Tensorflow dataset 불러오기 numpy를 이용해서 데이터 만들기 features, labels = (np.random.sample((100,2)), np.random.sample((100,1))) dataset =.. 2021. 5. 13. 이전 1 다음 728x90