Merge

Dockerfile

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

dlgssdpytorch.py

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