ignite.engine.supervised_training_step_amp(model, optimizer, loss_fn, device=None, ... Factory function for supervised training using torch.cuda.amp .
確定! 回上一頁