This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# Sketch-specific note: a roughly ~25 run battery for this code estimated a roughly ~93.11% accuracy in the same number of steps as the baseline network, ~1.7x runtime overhead (much of which goes to the torch.randn allocations and extra layer calculations). | |
# Note: The one change we need to make if we're in Colab is to uncomment this below block. | |
# If we are in an ipython session or a notebook, clear the state to avoid bugs | |
#""" | |
try: | |
_ = get_ipython().__class__.__name__ | |
## we set -f below to avoid prompting the user before clearing the notebook state | |
%reset -f | |
except NameError: |