Traceback (most recent call last):


 "C:\Users\jinyi\faceswap\lib\cli.py", line 107, in execute_script

process.process()


 "C:\Users\jinyi\faceswap\scripts\train.py", line 101, in process

self.end_thread(thread, err)


 "C:\Users\jinyi\faceswap\scripts\train.py", line 126, in end_thread

thread.join()


 "C:\Users\jinyi\faceswap\lib\multithreading.py", line 443, in join

raise thread.err[1].with_traceback(thread.err[2])


 "C:\Users\jinyi\faceswap\lib\multithreading.py", line 381, in run

self._target(*self._args, **self._kwargs)


 "C:\Users\jinyi\faceswap\scripts\train.py", line 152, in training

raise err


 "C:\Users\jinyi\faceswap\scripts\train.py", line 142, in training

self.run_training_cycle(model, trainer)


 "C:\Users\jinyi\faceswap\scripts\train.py", line 214, in run_training_cycle

trainer.train_one_step(viewer, timelapse)


 "C:\Users\jinyi\faceswap\plugins\train\trainer\_base.py", line 139, in train_one_step

loss[side] = batcher.train_one_batch(do_preview)


 "C:\Users\jinyi\faceswap\plugins\train\trainer\_base.py", line 214, in train_one_batch

loss = self.model.predictors[self.side].train_on_batch(*batch)


 "D:\PC_apps\Anaconda3\envs\faceswap\lib\site-packages\keras\engine\training.py", line 1217, in train_on_batch

outputs = self.train_function(ins)


 "D:\PC_apps\Anaconda3\envs\faceswap\lib\site-packages\keras\backend\tensorflow_backend.py", line 2715, in __call__

return self._call(inputs)


 "D:\PC_apps\Anaconda3\envs\faceswap\lib\site-packages\keras\backend\tensorflow_backend.py", line 2675, in _call

fetched = self._callable_fn(*array_vals)


 "D:\PC_apps\Anaconda3\envs\faceswap\lib\site-packages\tensorflow\python\client\session.py", line 1439, in __call__

run_metadata_ptr)


 "D:\PC_apps\Anaconda3\envs\faceswap\lib\site-packages\tensorflow\python\framework\errors_impl.py", line 528, in __exit__

c_api.TF_GetCode(self.status.status))
tensorflow.python.framework.errors_impl.ResourceExhaustedError: OOM when allocating tensor with shape[16384,1024] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc

Traceback (most recent call last):


 "main.py", line 69, in <module>

tf.app.run()


 "/usr/local/lib/python3.5/dist-packages/tensorflow/python/platform/app.py", line 125, in run

_sys.exit(main(argv))


 "main.py", line 55, in main

learner.train()


 "/opt/ml/env/learners/discr_channel_pruning/learner.py", line 145, in train

self.__choose_discr_chns()


 "/opt/ml/env/learners/discr_channel_pruning/learner.py", line 480, in __choose_discr_chns

self.sess_train.run(self.block_train_ops[idx_block])


 "/usr/local/lib/python3.5/dist-packages/tensorflow/python/client/session.py", line 929, in run

run_metadata_ptr)


 "/usr/local/lib/python3.5/dist-packages/tensorflow/python/client/session.py", line 1152, in _run

feed_dict_tensor, options, run_metadata)


 "/usr/local/lib/python3.5/dist-packages/tensorflow/python/client/session.py", line 1328, in _do_run

run_metadata)


 "/usr/local/lib/python3.5/dist-packages/tensorflow/python/client/session.py", line 1348, in _do_call

raise type(e)(node_def, op, message)
tensorflow.python.framework.errors_impl.ResourceExhaustedError: OOM when allocating tensor with shape[4096,512,7,7] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc
