weightmatrices/86755eee0ae1master
weightmatrices/
86755eee0ae1master
README.md
README.md
Creation for weight matrices for initialising the first layer of a neural network
Dependencies & Environment
- Conda needs to be installed
- Setup the conda environment weightmatrices by running:
` bash setup_dependencies.sh `
Usage
- To run the main script(s) that produce the weight matrices run:
` bash run.sh `
If you want to directly run main.py to use command line args (e.g. python main.py --methods 'pca' --nhidden 10 25), do not forget to activate the environment by running:
` conda activate weightmatrices `
Remarks
By default, Datasets are downloaded and put into a /datasets directory. To avoid this you can:
- Change the root variable in the torchvision data loader function.
- Create the /datasets folder and corresponding subfolder and put the datasets (or a link) in there.
c4science · Help