Traceback (most recent call last): File "C:\Users\yvesb\AppData\Roaming\Python\Python39\site-packages\tensorflow\python\client\session.py", line 1375, in _do_call return fn(*args) File "C:\Users\yvesb\AppData\Roaming\Python\Python39\site-packages\tensorflow\python\client\session.py", line 1359, in _run_fn return self._call_tf_sessionrun(options, feed_dict, fetch_list, File "C:\Users\yvesb\AppData\Roaming\Python\Python39\site-packages\tensorflow\python\client\session.py", line 1451, in _call_tf_sessionrun return tf_session.TF_SessionRun_wrapper(self._session, options, feed_dict, tensorflow.python.framework.errors_impl.ResourceExhaustedError: OOM when allocating tensor with shape[20,65536,64] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc [[{{node gradients/MatMul_grad/ArithmeticOptimizer/FoldTransposeIntoMatMul_MatMul_1}}]] Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info. During handling of the above exception, another exception occurred: Traceback (most recent call last): File "D:\Documents\tensorflow\fast-style-transfer-master2\fast-style-transfer-master\style.py", line 167, in main() File "D:\Documents\tensorflow\fast-style-transfer-master2\fast-style-transfer-master\style.py", line 147, in main for preds, losses, i, epoch in optimize(*args, **kwargs): File "src\optimize.py", line 114, in optimize train_step.run(feed_dict=feed_dict) File "C:\Users\yvesb\AppData\Roaming\Python\Python39\site-packages\tensorflow\python\framework\ops.py", line 2625, in run _run_using_default_session(self, feed_dict, self.graph, session) File "C:\Users\yvesb\AppData\Roaming\Python\Python39\site-packages\tensorflow\python\framework\ops.py", line 5578, in _run_using_default_session session.run(operation, feed_dict) File "C:\Users\yvesb\AppData\Roaming\Python\Python39\site-packages\tensorflow\python\client\session.py", line 967, in run result = self._run(None, fetches, feed_dict, options_ptr, File "C:\Users\yvesb\AppData\Roaming\Python\Python39\site-packages\tensorflow\python\client\session.py", line 1190, in _run results = self._do_run(handle, final_targets, final_fetches, File "C:\Users\yvesb\AppData\Roaming\Python\Python39\site-packages\tensorflow\python\client\session.py", line 1368, in _do_run return self._do_call(_run_fn, feeds, fetches, targets, options, File "C:\Users\yvesb\AppData\Roaming\Python\Python39\site-packages\tensorflow\python\client\session.py", line 1394, in _do_call raise type(e)(node_def, op, message) tensorflow.python.framework.errors_impl.ResourceExhaustedError: OOM when allocating tensor with shape[20,65536,64] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc [[{{node gradients/MatMul_grad/ArithmeticOptimizer/FoldTransposeIntoMatMul_MatMul_1}}]] Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info. PS D:\Documents\tensorflow\fast-style-transfer-master2\fast-style-transfer-master>