This PR moves the CNN model definition from train.py to a new file cnn.py. The CNN model is now defined in its own script, which improves code organization and modularity. Additionally, the necessary modules are imported in cnn.py to ensure the CNN model can be properly instantiated.
Summary of Changes
Created a new file cnn.py and moved the CNN class definition from train.py to cnn.py.
Removed the CNN class definition from train.py.
Added an import statement at the top of train.py to import the CNN model from cnn.py.
Ensured that cnn.py is in the same directory as train.py for the import statement to work correctly.
These changes make the codebase more maintainable and extensible by separating the CNN model from the training script and improving code organization.
Fixes #4.
To checkout this PR branch, run the following command in your terminal:
git checkout sweep/move-cnn-model
To get Sweep to edit this pull request, leave a comment below or in the code. Leaving a comment in the code will only modify the file but commenting below can change the entire PR.
Description
This PR moves the CNN model definition from
train.py
to a new filecnn.py
. The CNN model is now defined in its own script, which improves code organization and modularity. Additionally, the necessary modules are imported incnn.py
to ensure the CNN model can be properly instantiated.Summary of Changes
cnn.py
and moved the CNN class definition fromtrain.py
tocnn.py
.train.py
.train.py
to import the CNN model fromcnn.py
.cnn.py
is in the same directory astrain.py
for the import statement to work correctly.These changes make the codebase more maintainable and extensible by separating the CNN model from the training script and improving code organization.
Fixes #4.
To checkout this PR branch, run the following command in your terminal:
To get Sweep to edit this pull request, leave a comment below or in the code. Leaving a comment in the code will only modify the file but commenting below can change the entire PR.