Я пытаюсь использовать мои генераторы, которые работали с tf2.0.0, с tf2.1.0 и пользовательской функцией потерь. В принципе это не должно ничего менять, но я получаю странные ошибки:
WARNING:tensorflow:sample_weight modes were coerced from
...
to
['...']
2020-03-07 23:54:35.716875: E tensorflow/stream_executor/cuda/cuda_blas.cc:238] failed to create cublas handle: CUBLAS_STATUS_NOT_INITIALIZED
2020-03-07 23:54:35.716961: W tensorflow/stream_executor/stream.cc:2041] attempting to perform BLAS operation using StreamExecutor without BLAS support
2020-03-07 23:54:35.717112: W tensorflow/core/common_runtime/base_collective_executor.cc:217] BaseCollectiveExecutor::StartAbort Internal: BlasScal failed : in.shape=[2,256,256]
[[{{node loss/p_re_lu_15_loss/lambda_1/IFFT2D}}]]
Есть идеи?
Я использую генератор с model.fit_generator (), который выплевывает партии (X, Y )