---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
Cell In[28], line 12
      9 reduce_lr_loss = ReduceLROnPlateau(monitor='val_loss', cooldown=1, verbose=1, min_delta=0.00001, factor=0.1, patience=2, mode='min')
     10 callbacks = [early_stopping, mcp_save, reduce_lr_loss, csvlogger]
---> 12 conv_model = unet_2D(config)
     14 # FIT THE MODEL
     15 history = conv_model.fit(
     16     x=X_train,
     17     y=y_train,
   (...)
     22     callbacks=callbacks
     23 )

Cell In[25], line 51, in unet_2D(config)
     49 for jj in range(num_conv_per_block):
     50   first_block_input = final_input if(ii == 0 and jj == 0) else conv
---> 51   conv = L.Conv2D(nfmaps, filter_width, padding=padding, dilation_rate=dilation)(first_block_input)
     52   if(batch_norm): conv = L.BatchNormalization()(conv)
     53   conv = L.LeakyReLU()(conv) if(conv_activation == 'leaky_relu') else L.Activation(conv_activation)(conv)

File ~\anaconda3\lib\site-packages\keras\utils\traceback_utils.py:70, in filter_traceback.<locals>.error_handler(*args, **kwargs)
     67     filtered_tb = _process_traceback_frames(e.__traceback__)
     68     # To get the full stack trace, call:
     69     # `tf.debugging.disable_traceback_filtering()`
---> 70     raise e.with_traceback(filtered_tb) from None
     71 finally:
     72     del filtered_tb

File ~\anaconda3\lib\site-packages\keras\backend.py:2100, in RandomGenerator.random_uniform(self, shape, minval, maxval, dtype, nonce)
   2098     if nonce:
   2099         seed = tf.random.experimental.stateless_fold_in(seed, nonce)
-> 2100     return tf.random.stateless_uniform(
   2101         shape=shape,
   2102         minval=minval,
   2103         maxval=maxval,
   2104         dtype=dtype,
   2105         seed=seed,
   2106     )
   2107 return tf.random.uniform(
   2108     shape=shape,
   2109     minval=minval,
   (...)
   2112     seed=self.make_legacy_seed(),
   2113 )

ValueError: Memory growth cannot differ between GPU devices