TypeError in a tf file

[selingok033]

I have cloned the project and when I run it directly, I get the following error. I haven't been able to solve the problem yet, so if anyone can help, I would appreciate it if they could share their solution.

Traceback (most recent call last):
File "capsulenet.py", line 242, in
model, eval_model, manipulate_model = CapsNet(input_shape=x_train.shape[1:],
File "capsulenet.py", line 49, in CapsNet
digitcaps = CapsuleLayer(num_capsule=n_class, dim_capsule=16, routings=routings,
File "/storage/selin/miniconda3/envs/tf2/lib/python3.8/site-packages/tensorflow/python/keras/engine/base_layer.py", line 897, in call
self._maybe_build(inputs)
File "/storage/selin/miniconda3/envs/tf2/lib/python3.8/site-packages/tensorflow/python/keras/engine/base_layer.py", line 2416, in _maybe_build
self.build(input_shapes) # pylint:disable=not-callable
File "/storage/selin/Works/capsnet-keras/capsulelayers.py", line 113, in build
self.W = self.add_weight(shape=[self.num_capsule, self.input_num_capsule,
File "/storage/selin/miniconda3/envs/tf2/lib/python3.8/site-packages/tensorflow/python/keras/engine/base_layer.py", line 560, in add_weight
variable = self._add_variable_with_custom_getter(
File "/storage/selin/miniconda3/envs/tf2/lib/python3.8/site-packages/tensorflow/python/training/tracking/base.py", line 738, in _add_variable_with_custom_getter
new_variable = getter(
File "/storage/selin/miniconda3/envs/tf2/lib/python3.8/site-packages/tensorflow/python/keras/engine/base_layer_utils.py", line 129, in make_variable
return tf_variables.VariableV1(
File "/storage/selin/miniconda3/envs/tf2/lib/python3.8/site-packages/tensorflow/python/ops/variables.py", line 259, in call
return cls._variable_v1_call(*args, **kwargs)
File "/storage/selin/miniconda3/envs/tf2/lib/python3.8/site-packages/tensorflow/python/ops/variables.py", line 205, in _variable_v1_call
return previous_getter(
File "/storage/selin/miniconda3/envs/tf2/lib/python3.8/site-packages/tensorflow/python/ops/variables.py", line 198, in
previous_getter = lambda **kwargs: default_variable_creator(None, **kwargs)
File "/storage/selin/miniconda3/envs/tf2/lib/python3.8/site-packages/tensorflow/python/ops/variable_scope.py", line 2584, in default_variable_creator
return resource_variable_ops.ResourceVariable(
File "/storage/selin/miniconda3/envs/tf2/lib/python3.8/site-packages/tensorflow/python/ops/variables.py", line 263, in call
return super(VariableMetaclass, cls).call(*args, **kwargs)
File "/storage/selin/miniconda3/envs/tf2/lib/python3.8/site-packages/tensorflow/python/ops/resource_variable_ops.py", line 1423, in init
self._init_from_args(
File "/storage/selin/miniconda3/envs/tf2/lib/python3.8/site-packages/tensorflow/python/ops/resource_variable_ops.py", line 1567, in _init_from_args
initial_value() if init_from_fn else initial_value,
File "/storage/selin/miniconda3/envs/tf2/lib/python3.8/site-packages/tensorflow/python/keras/engine/base_layer_utils.py", line 121, in
init_val = lambda: initializer(shape, dtype=dtype)
File "/storage/selin/miniconda3/envs/tf2/lib/python3.8/site-packages/tensorflow/python/ops/init_ops_v2.py", line 545, in call
fan_in, fan_out = _compute_fans(scale_shape)
File "/storage/selin/miniconda3/envs/tf2/lib/python3.8/site-packages/tensorflow/python/ops/init_ops.py", line 1422, in _compute_fans
receptive_field_size *= dim
TypeError: unsupported operand type(s) for *=: 'int' and 'NoneType'