Jia_Ming:

for paras in model.parameters():
    paras.requires_grad = False



Hi,
on my first glance, I am not sure why you need this line.
Can you try removing this line or at least change the paras.requires_grad to True back, once you enter your first epoch?