Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Smt #45

Open
wants to merge 45 commits into
base: main
Choose a base branch
from
Open

Smt #45

wants to merge 45 commits into from

Conversation

dpaiton
Copy link
Owner

@dpaiton dpaiton commented Jan 28, 2021

No description provided.

* adds fastMNIST dataset and parameters
* fixes conv lca params to match between pytorch & tf1x - no longer
getting NAN
* fixes relative imports for utils/loaders.py - need to propagate to
other files
* fixes minor bugs in notebooks and run_utils
* improves some error messaging
removes incomplete aversarial_analysis script
all pytorch tests pass
adds convolutional MLP model with max pooling
reorders all expected datashapes to have channels first
moves typical log outputs to base model
adds optimizer to checkpoint writing
adds ability to load checkpoints from a log file
adds ability to boot from checkpoint at the start of training (untested)
adds ability to ignore gradients or include them when training ensemble models (untested)
datasets now include a 'num_pixels' parameter in their output
adds utility to laod parameters from a log file
adds preprocessing capability to FastMNIST
minor typo fixes
logging now includes computer environment and model architecture details
lca now outputs fraction nonzero for channel location as well as convolutional spatial map
fixes a checkpoint loading bug for ensemble models
fixes a bug with ensemble modules that caused submodules of the same type to clobber
removes flatten_feature_map function from utils.data_processing in favor of one-line option
reorganizes mlp and lca forward function calls for cleaner integration into ensembles
renames lca num_latent params to layer_channels to match mlp specification
adds pooling params to the test suite
changed some logging apis to be more intuitive / general
standardization can now use the dataset mean & std
dataset_utils now outputs original dataset mean & std
removes unnecessary thresholding operators from modules/activations
cleans up inhibitory connectivity function & adds convolutional version
switched to torch.mm instead of torch.matmul when matrices are 2D
layer_channels behaves like mlp, and now must include input channels
minor comment addition/removal in pooling_module
returned cifar preprocessing to be samplewise standardization
fixed bug in tests with new dataset outputs
updates ensemble lca test with comments & fixes ensemble state dict loading bug
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant