for i in range(epoch):
        train_set = get_train_set()
        inputs = [adj_matrix,adj_features,rel_features]

        optimizer.zero_grad()
        output = model(inputs)
        loss_train = criterion(output, train_set)
        loss_train.backward()
        optimizer.step()