adjust docker file for train and test
This commit is contained in:
parent
ef624e52fd
commit
87a2663e53
@ -11,5 +11,7 @@ WORKDIR /app
|
|||||||
|
|
||||||
COPY ./download.sh ./
|
COPY ./download.sh ./
|
||||||
COPY ./calculate.sh ./
|
COPY ./calculate.sh ./
|
||||||
|
COPY ./train.py ./
|
||||||
|
COPY ./test.py ./
|
||||||
|
|
||||||
CMD bash
|
CMD bash download.sh && bash calculate.sh && python train.py && python test.py
|
@ -2,4 +2,5 @@ pip install kaggle
|
|||||||
kaggle datasets download -d fanconic/skin-cancer-malignant-vs-benign
|
kaggle datasets download -d fanconic/skin-cancer-malignant-vs-benign
|
||||||
unzip skin-cancer-malignant-vs-benign.zip
|
unzip skin-cancer-malignant-vs-benign.zip
|
||||||
rm skin-cancer-malignant-vs-benign.zip
|
rm skin-cancer-malignant-vs-benign.zip
|
||||||
|
rm -r data
|
||||||
|
|
||||||
|
@ -1,3 +1,5 @@
|
|||||||
matplotlib==3.8.3
|
matplotlib==3.8.3
|
||||||
numpy==1.26.4
|
numpy==1.26.4
|
||||||
|
torchvision==0.17.1
|
||||||
|
torch==2.2.1
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user