diff --git a/Makefile b/Makefile index 6bd5958e..9910f1b6 100644 --- a/Makefile +++ b/Makefile @@ -24,6 +24,10 @@ uninstall: ## Uninstall docs: ## Render annotated HTML python ../../pylit/pylit.py -t ../../pylit/template_docs.html -d html -w labml_nn +pages: ## Copy to lab-ml site + @cd ../lab-ml.github.io; git pull + cp -r html/* ../lab-ml.github.io/ + help: ## Show this help. @fgrep -h "##" $(MAKEFILE_LIST) | fgrep -v fgrep | sed -e 's/\\$$//' | sed -e 's/##//' diff --git a/labml_nn/recurrent_highway_networks/__init__.py b/labml_nn/recurrent_highway_networks/__init__.py index 1a2ac381..f431e6d4 100644 --- a/labml_nn/recurrent_highway_networks/__init__.py +++ b/labml_nn/recurrent_highway_networks/__init__.py @@ -150,5 +150,5 @@ class RHN(Module): # Stack the outputs and states out = torch.stack(out) s = torch.stack(s) - + return out, s