vsay01 · January 22, 2020 23:23
diff --git a/checkpoint.py b/checkpoint.py
 checkpoint = {
            'epoch': epoch + 1,
            'valid_loss_min': valid_loss,
            'state_dict': model.state_dict(),
            'optimizer': optimizer.state_dict(),
        }
	checkpoint = {
	'epoch': epoch + 1,
	'valid_loss_min': valid_loss,
	'state_dict': model.state_dict(),
	'optimizer': optimizer.state_dict(),
	}