from torch.optim.lr_scheduler import StepLR. from torch.utils.data ... self.log('val_loss', loss, on_epoch=True, reduce_fx=torch.mean).
確定! 回上一頁