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 cnn.py and imported into train.py. This improves the modularity of the code and makes it easier to manage and update the model architecture in the future.
Summary of Changes
Created a new file cnn.py and copied 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 class from cnn.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.
Description
This PR moves the CNN model definition from the
train.pyfile to a new filecnn.py. The CNN class is now defined incnn.pyand imported intotrain.py. This improves the modularity of the code and makes it easier to manage and update the model architecture in the future.Summary of Changes
cnn.pyand copied the CNN class definition fromtrain.pytocnn.py.train.py.train.pyto import the CNN 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.