Fatal编程技术网
  • keras/
  • Pytork'；s相当于批次（）上的keras系列

Pytork'；s相当于批次（）上的keras系列

keras deep-learning pytorch

Pytork'；s相当于批次（）上的keras系列,keras,deep-learning,pytorch,semantic-segmentation,Keras,Deep Learning,Pytorch,Semantic Segmentation,这是我想转换为Pytork的代码的一部分，但我一直想知道Pytork中是否有作为批处理（）上的train_的train_的函数，或者Pytork中是否有任何可用于此目的的自定义函数 acc_loss = model.train_on_batch([support, smask, query], qmask) 该模型是一个自定义vgg16编码器和一个用于分段的解码器 support,smask,query,qmask ## these are all rank-4 tensors having

这是我想转换为Pytork的代码的一部分，但我一直想知道Pytork中是否有作为批处理（）上的train_的
train_的函数，或者Pytork中是否有任何可用于此目的的自定义函数


 acc_loss = model.train_on_batch([support, smask, query], qmask)


该模型是一个自定义vgg16编码器和一个用于分段的解码器


support,smask,query,qmask ## these are all rank-4 tensors having (batch_size,channel,height,width)