как хранить train_loss и valid_loss отдельно от epoch_loss?

Спасибо, @Rob, ваш код работает нормально, но в моем случае я возвращаю изображение из галереи и беру имя изображения с помощью кода:

let filename = url.lastPathComponent

Но этот код, отображающий расширение изображения как .JPG (в заглавной букве), но сервер не принимает расширения в capital письме, поэтому я изменил свой код как:

 let filename =  (path.lastPathComponent as NSString).lowercaseString

, и теперь мой код работает нормально.

Спасибо:)