Small framework fixes - Githubissues

ebagdasa / backdoors101

Backdoors Framework for Deep Learning and Federated Learning. A light-weight tool to conduct your research on backdoors.

MIT License

337 stars 82 forks source link

Small framework fixes #10

Closed DavidHidde closed 3 years ago

DavidHidde commented 3 years ago

Some small fixes for the framework:

Fix the PIPA task and complex synthesizer not working
Add the MultiMNIST Task
Disable shuffle for test dataset of MultiMNIST
Add the test_batch_size param for MNIST and MultiMNIST
Speed up ImageNet conisderably by adding more workers and pinning memory

DavidHidde commented 3 years ago

It would probably be best to squash commits for this PR, since I have some unneccesary commits in there from my own experimentation. If that's not possible, I'll start from scratch with the fixes so it can be in a single commit.

ebagdasa commented 3 years ago

@DavidHidde if you are interested in working more on this, let's chat some time! Happy to help with your research.