base code for pytorch distributed, add cyclegan

2020-08-07 09:48:09 +08:00
commit f7843de45d
32 changed files with 1444 additions and 0 deletions
--- a/environment.yml
+++ b/environment.yml
@@ -0,0 +1,22 @@
+name: raycv
+channels:
+  - pytorch
+  - conda-forge
+  - defaults
+dependencies:
+  - python=3.8
+  - numpy
+  - ipython
+  - tqdm
+  - pyyaml
+  - pytorch=1.6.*
+  - torchvision
+  - cudatoolkit=10.2
+  - ignite
+  - tensorboard
+  - omegaconf
+  - python-lmdb
+  - fire
+  # - opencv
+  # - jupyterlab
+