Runtime error on saving the module

2

can you try :

model.save(‘./checkpoint/epoch’+str(epoch)+‘.ckpt’)

?