Skip to content
Snippets Groups Projects
Commit f7165208 authored by s88711's avatar s88711
Browse files

Update TRAIN_MODEL_MIX_R18.py

parent f8a91abc
No related branches found
No related tags found
No related merge requests found
......@@ -22,7 +22,7 @@ Refernces: LFI-3 cnn.py
This traines a Normal Resnet18 with the a Kegal dataset "Early detection of 3D printing issues" MIXED with a Dataset
of a our on to Domain Shift the Dataset to fit our camera and printer.
Link to Original Dataset: https://www.kaggle.com/datasets/gauravduttakiit/early-detection-of-3d-printing-issues?select=train
Link to the Mixed Dataset: ??? TODO ???
Link to the Mixed Dataset: https://drive.google.com/drive/folders/1zktZCBSIpamPsIcyLlYEq5N0dgM1k0Zi?usp=drive_link
The Mixed Dataset contains approximately 10% more images of labeled fail and correct prints from the printer of Paul Judis
"""
......@@ -339,4 +339,4 @@ if __name__ == '__main__':
print(f'Finished Training, Loss: {sum(training_history[0])/len(training_history[0]):.4f}')
torch.save(model.state_dict(), model_save_PATH)
with open(training_history_PATH, 'w') as f:
json.dump(training_history, f)
\ No newline at end of file
json.dump(training_history, f)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment