__init__
function
loss
model
train
