我目前有一个使用data_x dim = [300000,60,5]的VAE。我想添加data_y dim = [300000,1],这是一个二进制条件。我将在下面发布我的模型代码(我使用自定义的train_step)。也许我的整个方法都是错的?as a `Model` with a custom `train_step` def __init__(self, encoder, decoder, **kwargs):
源代码:我想使用上述模型和公共mri数据(Brats)来制作一个模型来分割脑肿瘤。我决定采用转移学习的方法,所以用brats 2019年的数据对上述模型进行了训练。为了保持这些层除了最后两个层(请参阅下面的图像)之外,其他的层是不可训练的,
for layer in model.layers:但是我发现我无法访问VAE部件的最后一层(标记为图像的底部)。,因为model.layers不包括<e