#2. 데이터셋 가져오는 코드

HephaestusProject · Aug 11, 2020 · fec38da · fec38da
2 parents 8f5c8e0 + 2b77859
commit fec38da
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -1,4 +1,4 @@
 FROM pytorch/pytorch:1.6.0-cuda10.1-cudnn7-runtime
 ADD requirements.txt /workspace
 RUN pip install -r requirements.txt 
-COPY . /workspace
+COPY . /workspace
diff --git a/requirements.txt b/requirements.txt
@@ -1,3 +1,2 @@
-torch==1.5.0
-torchvision==0.6.0 
-pytorch-lightning==0.8.5
+pytorch-lightning==0.8.5
+omegaconf