Ошибка при обучении на WSL2, которой не было в Windows ⇐ Linux

[Anonymous]

Я пытался создать простую языковую модель, используя Python и тензорный поток, и обнаружил, что для правильного использования графического процессора мне нужно будет работать на WSL, конечно, без проблем. Хотя возникла новая проблема. Выдает следующую ошибку: TypeError: Layer.__init__() принимает 1 позиционный аргумент, но заданы 2, которые взяты из этой функции

Код: Выделить всё

class MyModel(tf.keras.Model):
def __init__(self, vocab_size, embedding_dim, rnn_units):
super().__init__(self)
self.embedding = tf.keras.layers.Embedding(vocab_size, embedding_dim)
self.gru = tf.keras.layers.GRU(rnn_units,
return_sequences=True,
return_state=True)
self.dense = tf.keras.layers.Dense(vocab_size)

def call(self, inputs, states=None, return_state=False, training=False):
x = inputs
x = self.embedding(x, training=training)
if states is None:
states = self.gru.get_initial_state(x)
x, states = self.gru(x, initial_state=states, training=training)
x = self.dense(x, training=training)

if return_state:
return x, states
else:
return x

Если я изменю строку super().__init__(self) на super().__init__(), это решит эту проблему, но создаст другую, и теперь он выдает следующую ошибку: {{function_node __wrapped__Pack_N_2_device_/job:localhost/replica:0/task:0/device:GPU:0}} Формы всех входных данных должны совпадать: значения[0].shape = [256,100,256] != значения[ 1].shape = [] [Op:Pack] name:
Мне интересно, сможет ли мне помочь кто-нибудь с опытом работы в TF более 20 минут.
п>

Подробнее здесь: https://stackoverflow.com/questions/783 ... on-windows