Skip to content

Instantly share code, notes, and snippets.

@ultrons

ultrons/library-element.py

Created December 4, 2020 13:27
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save ultrons/dce5c661e598950edf94454e49f7bf78 to your computer and use it in GitHub Desktop.
Save ultrons/dce5c661e598950edf94454e49f7bf78 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
library-element.py
import torch
import torch.nn as nn
import torch.multiprocessing as mp
import torch.distributed as dist
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment