danielcwq/init_params_func.py

Created November 28, 2021 05:39

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/danielcwq/2ea0e8a44b28696dd2445a57ae749eea.js"></script>
Save danielcwq/2ea0e8a44b28696dd2445a57ae749eea to your computer and use it in GitHub Desktop.

Download ZIP

Initialise Random Weights

Raw

init_params_func.py

	def init_params(size, std=1.0):
	return (torch.randn(size)*std).requires_grad_()
	weights = init_params((28*28,1))
	bias = init_params(1)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment