diff --git a/hw1/cs285/infrastructure/pytorch_util.py b/hw1/cs285/infrastructure/pytorch_util.py index bc7a4081..5b7b2e0a 100644 --- a/hw1/cs285/infrastructure/pytorch_util.py +++ b/hw1/cs285/infrastructure/pytorch_util.py @@ -47,6 +47,7 @@ def build_mlp( # TODO: return a MLP. This should be an instance of nn.Module # Note: nn.Sequential is an instance of nn.Module. + return MLP raise NotImplementedError