Я обучил модель сегментации в Pytorch и протестировал ее, чтобы получить оценку F1 на 0.93 на моем локальном компьютере (Windows, conda, CUDA 10.2, Pytorch 1.2). Однако при тестировании на сервере Linux показатель F1 упал до 0.3 (conda, CUDA 9.0, Pytorch 1.1). Я дважды проверил, что оба набора кода, файлы меток и наборы тестов были одинаковыми, и в моем коде нет «явной» случайной выборки (даже если это так, эффект не должен быть таким драматичным c).
Я попытался сделать следующее, чтобы решить мою проблему:
Я подозревал, что это может иметь какое-то отношение к версиям Pytorch, поэтому установил Pytorch 1.1 на мою машину Windows, чтобы соответствовать серверу, но получил тот же 0.93 балл. Затем я подумал, что это связано с самой CUDA, поэтому полностью исключил фактор GPU и сделал вывод на процессор на сервере, я все равно получил 0.3 на сервере.
Что может быть причиной такого огромного расхождения?