This PR moves the CNN class definition from the train.py file to a new file cnn.py. The CNN class is an important component of the codebase and separating it into its own file improves code organization and modularity. Additionally, the train.py file is updated to import the CNN class from the new file.
Summary of Changes
Created a new file cnn.py and moved the CNN class definition from train.py to this file.
Imported the CNN class in train.py to ensure the training script can still access it.
Updated the train.py file to remove the CNN class definition and import the class from cnn.py.
Fixes #4.
To checkout this PR branch, run the following command in your terminal:
git checkout sweep/move-cnn-class_1
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 class definition from the
train.pyfile to a new filecnn.py. The CNN class is an important component of the codebase and separating it into its own file improves code organization and modularity. Additionally, thetrain.pyfile is updated to import the CNN class from the new file.Summary of Changes
cnn.pyand moved the CNN class definition fromtrain.pyto this file.train.pyto ensure the training script can still access it.train.pyfile to remove the CNN class definition and import the class fromcnn.py.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.