AdamW (model.parameters(), lr=0.001) # Wrap the model in a PopTorch training wrapper poptorch_model = poptorch.trainingModel(model, options=opts, ...
確定! 回上一頁