ganguli-lab / grid-pattern-formation Goto Github PK

I tried to run pip install --upgrade tensorflow==2.0.0rc2 on Ubuntu 20.04 but this version is not available. I read online that tensorflow version 2.0.0rc2 is available in manylinux2010, is this what you are using? Thank you for your support.

I am trying to use inspect_model.ipynb without re-fitting the model. I updated weight_dir correctly, but get the following error:

`---------------------------------------------------------------------------
AttributeError Traceback (most recent call last)
~\AppData\Local\Temp\ipykernel_25532\1426264973.py in
7 # # Load example weights stored on github
8 weight_dir = options.save_dir + '/example_trained_weights.npy'
----> 9 load_trained_weights(model, trainer, weight_dir)

...\grid-pattern-formation\utils.py in load_trained_weights(model, trainer, weight_dir)
129 # Load weights from npy array
130 weights = np.load(weight_dir, allow_pickle=True)
--> 131 model.RNN.set_weights(weights)
132 print('Loaded trained weights.')
133

~\anaconda3\envs\gfp\lib\site-packages\torch\nn\modules\module.py in getattr(self, name)
1267 if name in modules:
1268 return modules[name]
-> 1269 raise AttributeError("'{}' object has no attribute '{}'".format(
1270 type(self).name, name))
1271

AttributeError: 'RNN' object has no attribute 'set_weights'`

Hi! I'm trying to reproduce this paper but grid cells aren't emerging in my implementation. Could you please release your code?

Thank you!