s426206/ium_426206
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge

Browse Source
This commit is contained in:
Jan Nowak 2021-05-07 20:19:24 +02:00
commit a356589903
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
Dockerfile
View File

@ -23,5 +23,8 @@ WORKDIR /app
COPY ./dlgssdpytorch.py ./ COPY ./dlgssdpytorch.py ./
RUN chmod +x dlgssdpytorch.py RUN chmod +x dlgssdpytorch.py
<<<<<<< HEAD
COPY ./create_dataset.py ./ COPY ./create_dataset.py ./
RUN chmod +x create_dataset.py RUN chmod +x create_dataset.py
=======
>>>>>>> 6f485d6db1dc0417e5862be40269d7075200ad57

2
dlgssdpytorch.py
View File

@ -112,4 +112,4 @@ for var_name in optimizer.state_dict():
print(var_name, "\t", optimizer.state_dict()[var_name]) print(var_name, "\t", optimizer.state_dict()[var_name])
print("Mean squared error for training: ", np.mean(losses)) print("Mean squared error for training: ", np.mean(losses))
print("Mean squared error for validating: ", np.mean(val_losses)) print("Mean squared error for validating: ", np.mean(val_losses))
torch.save(model, 'model.pt') torch.save(model, 'model.pt')