s452627/ium_452627
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix14

Browse Source
This commit is contained in:
Jakub Henyk 2023-05-07 18:34:53 +02:00
parent 3e53e29514
commit d07246f63f
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Dockerfile
View File

@ -28,5 +28,5 @@ RUN chmod +x ./train.py
ARG epochs=5 ARG epochs=5
RUN echo $epochs RUN echo $epochs
CMD python3 ./train.py epochs CMD python ./train.py epochs
#CMD python3 ./test.py #CMD python3 ./test.py

2
train.py
View File

@ -74,7 +74,7 @@ if __name__ == '__main__':
criterion = nn.CrossEntropyLoss() criterion = nn.CrossEntropyLoss()
optimizer = optim.SGD(net.parameters(), lr=0.001, momentum=0.9) optimizer = optim.SGD(net.parameters(), lr=0.001, momentum=0.9)
epochs = sys.argv[1] epochs = int(sys.argv[1])
print(epochs) print(epochs)
trainNet(trainloader, criterion, optimizer, int(float(epochs))) trainNet(trainloader, criterion, optimizer, int(float(epochs)))