This PR moves the CNN model definition from the train.py file to a new file cnn.py. The CNN class is now defined in the cnn.py file, making the code more modular and easier to maintain. Additionally, the CNN model is imported in the train.py file to ensure the code still functions correctly.
Summary of Changes
Created a new file cnn.py in the root directory of the repository.
Moved the CNN class definition from train.py to cnn.py.
Imported the CNN model from cnn.py in the train.py file.
Fixes #4.
To checkout this PR branch, run the following command in your terminal:
git checkout sweep/move-cnn
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 the
train.pyfile to a new filecnn.py. The CNN class is now defined in thecnn.pyfile, making the code more modular and easier to maintain. Additionally, the CNN model is imported in thetrain.pyfile to ensure the code still functions correctly.Summary of Changes
cnn.pyin the root directory of the repository.train.pytocnn.py.cnn.pyin thetrain.pyfile.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.