DeepFaceLab fork which provides IPython Notebook to use DFL with Google Colab

Overview

DFL-Colab — DeepFaceLab fork for Google Colab

This project provides you IPython Notebook to use DeepFaceLab with Google Colaboratory.
You can create your deepfake video even if you don’t have a powerful GPU on a desktop PC.

Keep in mind that working with a remote server imposes some restrictions.
Because of this, working with DFL-Colab is not so convenient as working with DeepFaceLab directly on your computer.

DFL-Colab uses DeepFaceLab and to understand the workflow, you should first understand the DeepFaceLab workflow.

Official Links

  • DFL-Colab Notebook: Link -- Works with latest DFL version
  • DFL-Colab Notebook 1.0: Link -- Works with DFL 1.0 version
  • DeepFaceLab Repository: Link
  • Youtube channel: Link

Community support

  • Telegram Chat (English/Russian): Link
  • MrDeepFakes Forum: Link
  • MrDeepFakes DFL-Colab Guide: Link
  • MrDeepFakes DeepFaceLab Guide: Link
  • Reddit — r/GifFakes/: Link
  • Reddit — r/SFWdeepfakes/: Link
Comments
  • SAEHD Training Failure

    SAEHD Training Failure

    Model Training (SAEHD and Quick96) fails upon execution.

    1. DLF installation functions
    2. Frame extraction functions
    3. DST and SRC face functions
    4. Xseg training functions
    5. Xseg apply/remove functions
    6. Model training fails
    • Sample error code

    Process Process-11: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last): File "/usr/lib/python3.8/multiprocessing/process.py", line 315, in _bootstrap self.run() File "/usr/lib/python3.8/multiprocessing/process.py", line 108, in run self._target(*self._args, **self._kwargs) File "/content/DeepFaceLab/core/joblib/SubprocessGenerator.py", line 54, in process_func gen_data = next (self.generator_func) File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 136, in batch_func raise Exception ("Exception occured in sample %s. Error: %s" % (sample.filename, traceback.format_exc() ) ) Exception: Exception occured in sample /content/workspace/data_src/aligned/00973.jpg. Error: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    Process Process-14: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last): File "/usr/lib/python3.8/multiprocessing/process.py", line 315, in _bootstrap self.run() File "/usr/lib/python3.8/multiprocessing/process.py", line 108, in run self._target(*self._args, **self._kwargs) File "/content/DeepFaceLab/core/joblib/SubprocessGenerator.py", line 54, in process_func gen_data = next (self.generator_func) File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 136, in batch_func raise Exception ("Exception occured in sample %s. Error: %s" % (sample.filename, traceback.format_exc() ) ) Exception: Exception occured in sample /content/workspace/data_src/aligned/00402.jpg. Error: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    Process Process-15: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last): File "/usr/lib/python3.8/multiprocessing/process.py", line 315, in _bootstrap self.run() File "/usr/lib/python3.8/multiprocessing/process.py", line 108, in run self._target(*self._args, **self._kwargs) File "/content/DeepFaceLab/core/joblib/SubprocessGenerator.py", line 54, in process_func gen_data = next (self.generator_func) File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 136, in batch_func raise Exception ("Exception occured in sample %s. Error: %s" % (sample.filename, traceback.format_exc() ) ) Exception: Exception occured in sample /content/workspace/data_src/aligned/01929.jpg. Error: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    Process Process-12: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last): File "/usr/lib/python3.8/multiprocessing/process.py", line 315, in _bootstrap self.run() File "/usr/lib/python3.8/multiprocessing/process.py", line 108, in run self._target(*self._args, **self._kwargs) File "/content/DeepFaceLab/core/joblib/SubprocessGenerator.py", line 54, in process_func gen_data = next (self.generator_func) File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 136, in batch_func raise Exception ("Exception occured in sample %s. Error: %s" % (sample.filename, traceback.format_exc() ) ) Exception: Exception occured in sample /content/workspace/data_src/aligned/00293.jpg. Error: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    Process Process-16: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last): File "/usr/lib/python3.8/multiprocessing/process.py", line 315, in _bootstrap self.run() File "/usr/lib/python3.8/multiprocessing/process.py", line 108, in run self._target(*self._args, **self._kwargs) File "/content/DeepFaceLab/core/joblib/SubprocessGenerator.py", line 54, in process_func gen_data = next (self.generator_func) File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 136, in batch_func raise Exception ("Exception occured in sample %s. Error: %s" % (sample.filename, traceback.format_exc() ) ) Exception: Exception occured in sample /content/workspace/data_src/aligned/00412.jpg. Error: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    Process Process-13: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last): File "/usr/lib/python3.8/multiprocessing/process.py", line 315, in _bootstrap self.run() File "/usr/lib/python3.8/multiprocessing/process.py", line 108, in run self._target(*self._args, **self._kwargs) File "/content/DeepFaceLab/core/joblib/SubprocessGenerator.py", line 54, in process_func gen_data = next (self.generator_func) File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 136, in batch_func raise Exception ("Exception occured in sample %s. Error: %s" % (sample.filename, traceback.format_exc() ) ) Exception: Exception occured in sample /content/workspace/data_src/aligned/02330.jpg. Error: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    Loading samples: 100% 308/308 [00:00<00:00, 678.14it/s] Process Process-26: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last): File "/usr/lib/python3.8/multiprocessing/process.py", line 315, in _bootstrap self.run() File "/usr/lib/python3.8/multiprocessing/process.py", line 108, in run self._target(*self._args, **self._kwargs) File "/content/DeepFaceLab/core/joblib/SubprocessGenerator.py", line 54, in process_func gen_data = next (self.generator_func) File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 136, in batch_func raise Exception ("Exception occured in sample %s. Error: %s" % (sample.filename, traceback.format_exc() ) ) Exception: Exception occured in sample /content/workspace/data_dst/aligned/00014_0.jpg. Error: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    Process Process-27: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last): File "/usr/lib/python3.8/multiprocessing/process.py", line 315, in _bootstrap self.run() File "/usr/lib/python3.8/multiprocessing/process.py", line 108, in run self._target(*self._args, **self._kwargs) File "/content/DeepFaceLab/core/joblib/SubprocessGenerator.py", line 54, in process_func gen_data = next (self.generator_func) File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 136, in batch_func raise Exception ("Exception occured in sample %s. Error: %s" % (sample.filename, traceback.format_exc() ) ) Exception: Exception occured in sample /content/workspace/data_dst/aligned/00108_0.jpg. Error: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    Process Process-25: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last): File "/usr/lib/python3.8/multiprocessing/process.py", line 315, in _bootstrap self.run() File "/usr/lib/python3.8/multiprocessing/process.py", line 108, in run self._target(*self._args, **self._kwargs) File "/content/DeepFaceLab/core/joblib/SubprocessGenerator.py", line 54, in process_func gen_data = next (self.generator_func) File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 136, in batch_func raise Exception ("Exception occured in sample %s. Error: %s" % (sample.filename, traceback.format_exc() ) ) Exception: Exception occured in sample /content/workspace/data_dst/aligned/00054_0.jpg. Error: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    Process Process-29: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last): File "/usr/lib/python3.8/multiprocessing/process.py", line 315, in _bootstrap self.run() File "/usr/lib/python3.8/multiprocessing/process.py", line 108, in run self._target(*self._args, **self._kwargs) File "/content/DeepFaceLab/core/joblib/SubprocessGenerator.py", line 54, in process_func gen_data = next (self.generator_func) File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 136, in batch_func raise Exception ("Exception occured in sample %s. Error: %s" % (sample.filename, traceback.format_exc() ) ) Exception: Exception occured in sample /content/workspace/data_dst/aligned/00276_0.jpg. Error: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    Process Process-30: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last): File "/usr/lib/python3.8/multiprocessing/process.py", line 315, in _bootstrap self.run() File "/usr/lib/python3.8/multiprocessing/process.py", line 108, in run self._target(*self._args, **self._kwargs) File "/content/DeepFaceLab/core/joblib/SubprocessGenerator.py", line 54, in process_func gen_data = next (self.generator_func) File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 136, in batch_func raise Exception ("Exception occured in sample %s. Error: %s" % (sample.filename, traceback.format_exc() ) ) Exception: Exception occured in sample /content/workspace/data_dst/aligned/00209_0.jpg. Error: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    Process Process-28: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last): File "/usr/lib/python3.8/multiprocessing/process.py", line 315, in _bootstrap self.run() File "/usr/lib/python3.8/multiprocessing/process.py", line 108, in run self._target(*self._args, **self._kwargs) File "/content/DeepFaceLab/core/joblib/SubprocessGenerator.py", line 54, in process_func gen_data = next (self.generator_func) File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 136, in batch_func raise Exception ("Exception occured in sample %s. Error: %s" % (sample.filename, traceback.format_exc() ) ) Exception: Exception occured in sample /content/workspace/data_dst/aligned/00189_0.jpg. Error: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    opened by fulanito3030 6
  • pretrain faceset download error

    pretrain faceset download error

    I am trying to run the first step on the DFL colab latest notebook but get this error on the last part of it. DeepFaceLab uses faceset.pak which could simplify things, not needing to unzip etc.

    I'll try to figure it out locally and create a pull request

    [/content/pretrain_CelebA.zip]
      End-of-central-directory signature not found.  Either this file is not
      a zipfile, or it constitutes one disk of a multi-part archive.  In the
      latter case the central directory and zipfile comment will be found on
      the last disk(s) of this archive.
    unzip:  cannot find zipfile directory in one of /content/pretrain_CelebA.zip or
            /content/pretrain_CelebA.zip.zip, and cannot find /content/pretrain_CelebA.zip.ZIP, period.
    
    opened by camjac251 6
  • TypeError: object of type 'PosixPath' has no len()

    TypeError: object of type 'PosixPath' has no len()

    OS: Debian Linux GPU: Nvidia T4 CUDA: 10.0.130-1 Python: 3.6 Command line: python3.6 DeepFaceLab/main.py merge --input-dir workspace/data_dst --output-dir workspace/data_dst/merged --output-mask-dir workspace/data_dst/merged_mask --aligned-dir workspace/data_dst/aligned --model-dir workspace/model --model "SAEHD"

    I've been training a model for the past 48 hours and I have already merged the model and exported it to a video twice with no problems. I just tried to merge the model again and now it gets to frame 95 and crashes. See the error log below this paragraph. Is my model useless now? I've been training it on a Google Compute VM so if the model is useless then I've wasted $10 on the Google Compute VM so far.

    Merging: 3% 95/3623 [00:05<03:27, 17.00it/s]Exception while process data [/content/workspace/data_dst/00096.jpg]: Traceback (most recent call last): File "/content/DeepFaceLab/mainscripts/Merger.py", line 151, in process_data final_img = MergeMasked (self.predictor_func, self.predictor_input_shape, cfg, frame_info) File "/content/DeepFaceLab/merger/MergeMasked.py", line 327, in MergeMasked out_img, out_img_merging_mask = MergeMaskedFace (predictor_func, predictor_input_shape, cfg, frame_info, img_bgr_uint8, img_bgr, img_landmarks) File "/content/DeepFaceLab/merger/MergeMasked.py", line 13, in MergeMaskedFace img_face_mask_a = LandmarksProcessor.get_image_hull_mask (img_bgr.shape, img_face_landmarks) File "/content/DeepFaceLab/facelib/LandmarksProcessor.py", line 355, in get_image_hull_mask lmrks = expand_eyebrows(image_landmarks, eyebrows_expand_mod) File "/content/DeepFaceLab/facelib/LandmarksProcessor.py", line 324, in expand_eyebrows if len(lmrks) != 68: TypeError: object of type 'PosixPath' has no len()

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last): File "/content/DeepFaceLab/core/joblib/SubprocessorBase.py", line 71, in _subprocess_run result = self.process_data (data) File "/content/DeepFaceLab/mainscripts/Merger.py", line 157, in process_data raise Exception( f'Error while merging file [{filepath}]: {e_str}' ) Exception: Error while merging file [/content/workspace/data_dst/00096.jpg]: Traceback (most recent call last): File "/content/DeepFaceLab/mainscripts/Merger.py", line 151, in process_data final_img = MergeMasked (self.predictor_func, self.predictor_input_shape, cfg, frame_info) File "/content/DeepFaceLab/merger/MergeMasked.py", line 327, in MergeMasked out_img, out_img_merging_mask = MergeMaskedFace (predictor_func, predictor_input_shape, cfg, frame_info, img_bgr_uint8, img_bgr, img_landmarks) File "/content/DeepFaceLab/merger/MergeMasked.py", line 13, in MergeMaskedFace img_face_mask_a = LandmarksProcessor.get_image_hull_mask (img_bgr.shape, img_face_landmarks) File "/content/DeepFaceLab/facelib/LandmarksProcessor.py", line 355, in get_image_hull_mask lmrks = expand_eyebrows(image_landmarks, eyebrows_expand_mod) File "/content/DeepFaceLab/facelib/LandmarksProcessor.py", line 324, in expand_eyebrows if len(lmrks) != 68: TypeError: object of type 'PosixPath' has no len()

    /usr/lib/python3.6/multiprocessing/semaphore_tracker.py:143: UserWarning: semaphore_tracker: There appear to be 1 leaked semaphores to clean up at shutdown len(cache)) Exception while process data [/content/workspace/data_dst/00097.jpg]: Traceback (most recent call last): File "/content/DeepFaceLab/mainscripts/Merger.py", line 151, in process_data final_img = MergeMasked (self.predictor_func, self.predictor_input_shape, cfg, frame_info) File "/content/DeepFaceLab/merger/MergeMasked.py", line 327, in MergeMasked out_img, out_img_merging_mask = MergeMaskedFace (predictor_func, predictor_input_shape, cfg, frame_info, img_bgr_uint8, img_bgr, img_landmarks) File "/content/DeepFaceLab/merger/MergeMasked.py", line 13, in MergeMaskedFace img_face_mask_a = LandmarksProcessor.get_image_hull_mask (img_bgr.shape, img_face_landmarks) File "/content/DeepFaceLab/facelib/LandmarksProcessor.py", line 355, in get_image_hull_mask lmrks = expand_eyebrows(image_landmarks, eyebrows_expand_mod) File "/content/DeepFaceLab/facelib/LandmarksProcessor.py", line 324, in expand_eyebrows if len(lmrks) != 68: TypeError: object of type 'PosixPath' has no len()

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last): File "/content/DeepFaceLab/core/joblib/SubprocessorBase.py", line 71, in _subprocess_run result = self.process_data (data) File "/content/DeepFaceLab/mainscripts/Merger.py", line 157, in process_data raise Exception( f'Error while merging file [{filepath}]: {e_str}' ) Exception: Error while merging file [/content/workspace/data_dst/00097.jpg]: Traceback (most recent call last): File "/content/DeepFaceLab/mainscripts/Merger.py", line 151, in process_data final_img = MergeMasked (self.predictor_func, self.predictor_input_shape, cfg, frame_info) File "/content/DeepFaceLab/merger/MergeMasked.py", line 327, in MergeMasked out_img, out_img_merging_mask = MergeMaskedFace (predictor_func, predictor_input_shape, cfg, frame_info, img_bgr_uint8, img_bgr, img_landmarks) File "/content/DeepFaceLab/merger/MergeMasked.py", line 13, in MergeMaskedFace img_face_mask_a = LandmarksProcessor.get_image_hull_mask (img_bgr.shape, img_face_landmarks) File "/content/DeepFaceLab/facelib/LandmarksProcessor.py", line 355, in get_image_hull_mask lmrks = expand_eyebrows(image_landmarks, eyebrows_expand_mod) File "/content/DeepFaceLab/facelib/LandmarksProcessor.py", line 324, in expand_eyebrows if len(lmrks) != 68: TypeError: object of type 'PosixPath' has no len()

    /usr/lib/python3.6/multiprocessing/semaphore_tracker.py:143: UserWarning: semaphore_tracker: There appear to be 1 leaked semaphores to clean up at shutdown len(cache))

    opened by l0ophole 6
  • Google Colab Auto Deletes

    Google Colab Auto Deletes "DeepFaceLab" Folder

    Merry Christmas, lovely fellow nerds.

    Is it just me, or is Google Colab on purpose messing with us?

    I'm paid Premium++ and it seems they even mess with their paying customers....

    It happened a few times in a row today...That the "DeepFaceLab" folder got deleted instantly after installation and my script errors out "No folder named 'content/DeepFaceLab'" etc.

    And yesterday a few times it "stopped" during the Extraction of frames from a video. With no explanation whatsoever. The random stops during Frame Extraction were not a RAM or Harddisk space problem. The full video is cut into 10-minute parts. I don't even use any GPU for extracting frames so I don't understand why they seem to sabotage my work.

    Let's assume they delete "DeepFaceLab" on purpose, for a moment: They don't know if I'm using DeepFaceLab for company productions or hobby projects. How could they defend deleting "DeepFaceLab" the second after I installed it?

    • What is your experience friends and what are your thoughts? 🤔

    @chervonij

    opened by qkum 5
  • Error: module 'tensorflow' has no attribute 'ConfigProto'

    Error: module 'tensorflow' has no attribute 'ConfigProto'

    Have started getting error:

    [CPU] : CPU [0] : Tesla V100-SXM2-16GB

    [0] Which GPU indexes to choose? : 0

    Error: module 'tensorflow' has no attribute 'ConfigProto' Traceback (most recent call last): File "/content/DeepFaceLab/mainscripts/Trainer.py", line 55, in trainerThread debug=debug, File "/content/DeepFaceLab/models/ModelBase.py", line 151, in init nn.initialize(self.device_config) File "/content/DeepFaceLab/core/leras/nn.py", line 101, in initialize config = tf.ConfigProto() AttributeError: module 'tensorflow' has no attribute 'ConfigProto'

    when using %cd ./DeepFaceLab !git checkout d1a5639e9083508207962f9bd1392af145ad7c88

    using that commit used to work fine a few months ago.

    opened by tania010 5
  • Colab disconnected in just 40 minutes

    Colab disconnected in just 40 minutes

    Hello, as I understand it, Colab must let me work for 12 hours, so DFL_Colab makes a backup every 11 hours. Well, only the first day I used Colab allowed me to work 12 hours. Now it goes offline due to Colab usage limits in under 45 minutes.

    The main problem is that when Colab disconnects, the workspace folder disappears and I cannot find a way to make a copy to drive of what has already been worked.

    Somebody help me please? Sorry for my English but I am Spanish.

    opened by reimond16 5
  • ModuleNotFoundError: No module named 'tensorflow.compat.v2

    ModuleNotFoundError: No module named 'tensorflow.compat.v2

    im facing issue with DFL_Colab_1-0.ipynb (because im using amd in my physical pc so i choose DFL_Colab_1-0.ipynb in colab)

    Using TensorFlow backend. Error: No module named 'tensorflow.compat.v2' Traceback (most recent call last): File "/content/DeepFaceLab/mainscripts/Trainer.py", line 52, in trainerThread device_args=device_args) File "/content/DeepFaceLab/models/ModelBase.py", line 142, in init nnlib.import_all(self.device_config) File "/content/DeepFaceLab/nnlib/nnlib.py", line 1613, in import_all nnlib.import_keras(device_config) File "/content/DeepFaceLab/nnlib/nnlib.py", line 229, in import_keras import keras as keras_ File "/usr/local/lib/python3.7/dist-packages/keras/init.py", line 3, in from . import utils File "/usr/local/lib/python3.7/dist-packages/keras/utils/init.py", line 27, in from .multi_gpu_utils import multi_gpu_model File "/usr/local/lib/python3.7/dist-packages/keras/utils/multi_gpu_utils.py", line 7, in from ..layers.merge import concatenate File "/usr/local/lib/python3.7/dist-packages/keras/layers/init.py", line 4, in from ..engine.base_layer import Layer File "/usr/local/lib/python3.7/dist-packages/keras/engine/init.py", line 3, in from .input_layer import Input File "/usr/local/lib/python3.7/dist-packages/keras/engine/input_layer.py", line 7, in from .base_layer import Layer File "/usr/local/lib/python3.7/dist-packages/keras/engine/base_layer.py", line 12, in from .. import initializers File "/usr/local/lib/python3.7/dist-packages/keras/initializers/init.py", line 17, in import tensorflow.compat.v2 as tf ModuleNotFoundError: No module named 'tensorflow.compat.v2'

    please help

    opened by hargums 4
  • imgaug / albumentations error

    imgaug / albumentations error

    Hi,

    I try to run Install or update DeepFaceLab from Github script but get this error

    ERROR: albumentations 0.1.12 has requirement imgaug<0.2.7,>=0.2.5, but you'll have imgaug 0.2.9 which is incompatible.
    
    opened by Soneji 4
  • ZeroDivisionError: integer division or modulo by zero

    ZeroDivisionError: integer division or modulo by zero

    Running Latest Colab gives following exception while Training Model (Step 4).

    Traceback (most recent call last): File "/content/DeepFaceLab/mainscripts/Trainer.py", line 49, in trainerThread device_args=device_args) File "/content/DeepFaceLab/models/ModelBase.py", line 208, in init self.sample_for_preview = self.generate_next_sample() File "/content/DeepFaceLab/models/ModelBase.py", line 398, in generate_next_sample return [next(generator) for generator in self.generator_list] File "/content/DeepFaceLab/models/ModelBase.py", line 398, in return [next(generator) for generator in self.generator_list] File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 58, in next generator = self.generators[self.generator_counter % len(self.generators) ] ZeroDivisionError: integer division or modulo by zero Done.

    I had it working before but not sure what has changed.

    opened by somilhyperanna 4
  • Questions about usage

    Questions about usage

    So I started the training last night and closed the browser. After 10 minutes or so, I open the browser again and saw that instance was still running then I closed the browser again. Next day I check the instance only to find out that instance is terminated and no models have been uploaded to the drive.

    What is the correct usage here? Should I keep the browser open for 11 hours? Why model hasn't been uploaded to the drive?

    opened by Greywolf22 4
  • Getting ValueError: operands could not be broadcast together with shapes (144,144,3) (72,72,3) error

    Getting ValueError: operands could not be broadcast together with shapes (144,144,3) (72,72,3) error

    Tried this today and got the error below

    Drive already mounted at /content/drive; to attempt to forcibly remount, call drive.mount("/content/drive", force_remount=True).
    Time to backup: 10 hours
    /content
    Running trainer.
    
    Loading model...
    
    Model first run. Enter model options as default for each run.
    Write preview history? (y/n ?:help skip:n) : y
    Target iteration (skip:unlimited/default) : 
    0
    Batch_size (?:help skip:0) : 
    0
    Feed faces to network sorted by yaw? (y/n ?:help skip:n) : n
    Flip faces randomly? (y/n ?:help skip:y) : n
    Src face scale modifier % ( -30...30, ?:help skip:0) : 
    0
    Resolution ( 64-256 ?:help skip:128) : 144
    Half or Full face? (h/f, ?:help skip:f) : 
    f
    Learn mask? (y/n, ?:help skip:y) : 
    y
    Optimizer mode? ( 1,2,3 ?:help skip:1) : 
    1
    AE architecture (df, liae ?:help skip:df) : 
    df
    AutoEncoder dims (32-1024 ?:help skip:512) : 
    512
    Encoder dims per channel (21-85 ?:help skip:42) : 
    42
    Decoder dims per channel (10-85 ?:help skip:21) : 
    21
    Remove gray border? (y/n, ?:help skip:n) : 
    n
    Use multiscale decoder? (y/n, ?:help skip:n) : y
    Use pixel loss? (y/n, ?:help skip: n ) : 
    n
    Face style power ( 0.0 .. 100.0 ?:help skip:0.00) : 
    0.0
    Background style power ( 0.0 .. 100.0 ?:help skip:0.00) : 
    0.0
    Using TensorFlow backend.
    WARNING:tensorflow:From /usr/local/lib/python3.6/dist-packages/tensorflow/python/framework/op_def_library.py:263: colocate_with (from tensorflow.python.framework.ops) is deprecated and will be removed in a future version.
    Instructions for updating:
    Colocations handled automatically by placer.
    WARNING:tensorflow:From /usr/local/lib/python3.6/dist-packages/tensorflow/python/ops/math_ops.py:3066: to_int32 (from tensorflow.python.ops.math_ops) is deprecated and will be removed in a future version.
    Instructions for updating:
    Use tf.cast instead.
    Loading: 100% 6791/6791 [00:06<00:00, 1049.77it/s]
    Loading: 100% 6791/6791 [00:05<00:00, 1145.45it/s]
    ===== Model summary =====
    == Model name: SAE
    ==
    == Current iteration: 0
    ==
    == Model options:
    == |== write_preview_history : True
    == |== batch_size : 4
    == |== sort_by_yaw : False
    == |== random_flip : False
    == |== resolution : 144
    == |== face_type : f
    == |== learn_mask : True
    == |== optimizer_mode : 1
    == |== archi : df
    == |== ae_dims : 512
    == |== e_ch_dims : 42
    == |== d_ch_dims : 21
    == |== remove_gray_border : False
    == |== multiscale_decoder : True
    == |== pixel_loss : False
    == |== face_style_power : 0.0
    == |== bg_style_power : 0.0
    == Running on:
    == |== [0 : Tesla T4]
    =========================
    Starting. Press "Enter" to stop training and save model.
    Error: operands could not be broadcast together with shapes (144,144,3) (72,72,3) 
    Traceback (most recent call last):
      File "/content/DeepFaceLab/mainscripts/Trainer.py", line 93, in trainerThread
        iter, iter_time = model.train_one_iter()
      File "/content/DeepFaceLab/models/ModelBase.py", line 376, in train_one_iter
        plist += [ (self.get_previews()[0][1], self.get_strpath_storage_for_file('preview.jpg') ) ]
      File "/content/DeepFaceLab/models/ModelBase.py", line 274, in get_previews
        return self.onGetPreview ( self.last_sample )
      File "/content/DeepFaceLab/models/Model_SAE/Model.py", line 411, in onGetPreview
        ar = S[i]*test_S_m[i], SS[i], D[i]*test_D_m[i], DD[i]*DDM[i], SD[i]*(DDM[i]*SDM[i])
    ValueError: operands could not be broadcast together with shapes (144,144,3) (72,72,3) 
    Done.
    /usr/lib/python3.6/multiprocessing/semaphore_tracker.py:143: UserWarning: semaphore_tracker: There appear to be 1 leaked semaphores to clean up at shutdown
      len(cache))
    /usr/lib/python3.6/multiprocessing/semaphore_tracker.py:143: UserWarning: semaphore_tracker: There appear to be 1 leaked semaphores to clean up at shutdown
      len(cache))
    /usr/lib/python3.6/multiprocessing/semaphore_tracker.py:143: UserWarning: semaphore_tracker: There appear to be 1 leaked semaphores to clean up at shutdown
      len(cache))
    /usr/lib/python3.6/multiprocessing/semaphore_tracker.py:143: UserWarning: semaphore_tracker: There appear to be 1 leaked semaphores to clean up at shutdown
      len(cache))
    
    opened by Greywolf22 4
  • Wording/Phrasing changes, Remove References to multiple Drive accounts, Rename function name in Import Workspace, Refactor Dataset/Model Download

    Wording/Phrasing changes, Remove References to multiple Drive accounts, Rename function name in Import Workspace, Refactor Dataset/Model Download

    I saw after the change in Google Drive (same account restriction), that a reference to multiple Google accounts was in the text box, so I ended up removing it but then wanted to go further and try helping with some of the other stuff as well. I updated the GPUs that Colab has as well, filled out the VRAM amount for the GPUs I have gotten. I also changed the function name in "import workspace". Also refactored the dataset and model downloading to hopefully make it easier in the future for multiple options and also allow it to be optional and not required.

    opened by camjac251 10
Releases(pretrain_GenericFFHQ)
Official Implementation of 'UPDeT: Universal Multi-agent Reinforcement Learning via Policy Decoupling with Transformers' ICLR 2021(spotlight)

UPDeT Official Implementation of UPDeT: Universal Multi-agent Reinforcement Learning via Policy Decoupling with Transformers (ICLR 2021 spotlight) The

hhhusiyi 96 Dec 22, 2022
custom pytorch implementation of MoCo v3

MoCov3-pytorch custom implementation of MoCov3 [arxiv]. I made minor modifications based on the official MoCo repository [github]. No ViT part code an

39 Nov 14, 2022
The dataset of tweets pulling from Twitters with keyword: Hydroxychloroquine, location: US, Time: 2020

HCQ_Tweet_Dataset: FREE to Download. Keywords: HCQ, hydroxychloroquine, tweet, twitter, COVID-19 This dataset is associated with the paper "Understand

2 Mar 16, 2022
Experiments for distributed optimization algorithms

Network-Distributed Algorithm Experiments -- This repository contains a set of optimization algorithms and objective functions, and all code needed to

Boyue Li 40 Dec 04, 2022
Automated detection of anomalous exoplanet transits in light curve data.

Automatically detecting anomalous exoplanet transits This repository contains the source code for the paper "Automatically detecting anomalous exoplan

1 Feb 01, 2022
A DCGAN to generate anime faces using custom mined dataset

Anime-Face-GAN-Keras A DCGAN to generate anime faces using custom dataset in Keras. Dataset The dataset is created by crawling anime database websites

Pavitrakumar P 190 Jan 03, 2023
Federated learning on graph, especially on graph neural networks (GNNs), knowledge graph, and private GNN.

Federated learning on graph, especially on graph neural networks (GNNs), knowledge graph, and private GNN.

keven 198 Dec 20, 2022
[ICCV 2021] Group-aware Contrastive Regression for Action Quality Assessment

CoRe Created by Xumin Yu*, Yongming Rao*, Wenliang Zhao, Jiwen Lu, Jie Zhou This is the PyTorch implementation for ICCV paper Group-aware Contrastive

Xumin Yu 31 Dec 24, 2022
[CVPR 2021] MiVOS - Mask Propagation module. Reproduced STM (and better) with training code :star2:. Semi-supervised video object segmentation evaluation.

MiVOS (CVPR 2021) - Mask Propagation Ho Kei Cheng, Yu-Wing Tai, Chi-Keung Tang [arXiv] [Paper PDF] [Project Page] [Papers with Code] This repo impleme

Rex Cheng 106 Jan 03, 2023
Certis - Certis, A High-Quality Backtesting Engine

Certis - Backtesting For y'all Certis is a powerful, lightweight, simple backtes

Yeachan-Heo 46 Oct 30, 2022
《Fst Lerning of Temporl Action Proposl vi Dense Boundry Genertor》(AAAI 2020)

Update 2020.03.13: Release tensorflow-version and pytorch-version DBG complete code. 2019.11.12: Release tensorflow-version DBG inference code. 2019.1

Tencent 338 Dec 16, 2022
One-Shot Neural Ensemble Architecture Search by Diversity-Guided Search Space Shrinking

One-Shot Neural Ensemble Architecture Search by Diversity-Guided Search Space Shrinking This is an official implementation for NEAS presented in CVPR

Multimedia Research 19 Sep 08, 2022
An implementation of an abstract algebra for music tones (pitches).

nbdev template Use this template to more easily create your nbdev project. If you are using an older version of this template, and want to upgrade to

Open Music Kit 0 Oct 10, 2022
Tensorflow implementation of soft-attention mechanism for video caption generation.

SA-tensorflow Tensorflow implementation of soft-attention mechanism for video caption generation. An example of soft-attention mechanism. The attentio

Paul Chen 153 Nov 14, 2022
Official implementation of Self-supervised Graph Attention Networks (SuperGAT), ICLR 2021.

SuperGAT Official implementation of Self-supervised Graph Attention Networks (SuperGAT). This model is presented at How to Find Your Friendly Neighbor

Dongkwan Kim 127 Dec 28, 2022
Object detection, 3D detection, and pose estimation using center point detection:

Objects as Points Object detection, 3D detection, and pose estimation using center point detection: Objects as Points, Xingyi Zhou, Dequan Wang, Phili

Xingyi Zhou 6.7k Jan 03, 2023
Learning to Reconstruct 3D Manhattan Wireframes from a Single Image

Learning to Reconstruct 3D Manhattan Wireframes From a Single Image This repository contains the PyTorch implementation of the paper: Yichao Zhou, Hao

Yichao Zhou 50 Dec 27, 2022
Open standard for machine learning interoperability

Open Neural Network Exchange (ONNX) is an open ecosystem that empowers AI developers to choose the right tools as their project evolves. ONNX provides

Open Neural Network Exchange 13.9k Dec 30, 2022
A library built upon PyTorch for building embeddings on discrete event sequences using self-supervision

pytorch-lifestream a library built upon PyTorch for building embeddings on discrete event sequences using self-supervision. It can process terabyte-si

Dmitri Babaev 103 Dec 17, 2022
ECLARE: Extreme Classification with Label Graph Correlations

ECLARE ECLARE: Extreme Classification with Label Graph Correlations @InProceedings{Mittal21b, author = "Mittal, A. and Sachdeva, N. and Agrawal

Extreme Classification 35 Nov 06, 2022