Spaces:
Sleeping
Sleeping
Zai
commited on
Commit
·
bbddfb0
1
Parent(s):
34253f7
initializing parameters
Browse files- .github/workflows/python-package-conda.yml +0 -34
- .github/workflows/test.yaml +25 -0
- README.md +6 -6
- todo.txt +1 -0
- vegans/discriminator.py +10 -0
- vegans/generator.py +10 -0
- vegans/utils.py +1 -0
- vegans/vegans.py +1 -0
.github/workflows/python-package-conda.yml
DELETED
@@ -1,34 +0,0 @@
|
|
1 |
-
name: Python Package using Conda
|
2 |
-
|
3 |
-
on: [push]
|
4 |
-
|
5 |
-
jobs:
|
6 |
-
build-linux:
|
7 |
-
runs-on: ubuntu-latest
|
8 |
-
strategy:
|
9 |
-
max-parallel: 5
|
10 |
-
|
11 |
-
steps:
|
12 |
-
- uses: actions/checkout@v3
|
13 |
-
- name: Set up Python 3.10
|
14 |
-
uses: actions/setup-python@v3
|
15 |
-
with:
|
16 |
-
python-version: '3.10'
|
17 |
-
- name: Add conda to system path
|
18 |
-
run: |
|
19 |
-
# $CONDA is an environment variable pointing to the root of the miniconda directory
|
20 |
-
echo $CONDA/bin >> $GITHUB_PATH
|
21 |
-
- name: Install dependencies
|
22 |
-
run: |
|
23 |
-
conda env update --file environment.yml --name base
|
24 |
-
- name: Lint with flake8
|
25 |
-
run: |
|
26 |
-
conda install flake8
|
27 |
-
# stop the build if there are Python syntax errors or undefined names
|
28 |
-
flake8 . --count --select=E9,F63,F7,F82 --show-source --statistics
|
29 |
-
# exit-zero treats all errors as warnings. The GitHub editor is 127 chars wide
|
30 |
-
flake8 . --count --exit-zero --max-complexity=10 --max-line-length=127 --statistics
|
31 |
-
- name: Test with pytest
|
32 |
-
run: |
|
33 |
-
conda install pytest
|
34 |
-
pytest
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
.github/workflows/test.yaml
CHANGED
@@ -0,0 +1,25 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
name: Run Python Tests
|
2 |
+
|
3 |
+
on:
|
4 |
+
push:
|
5 |
+
branches:
|
6 |
+
- main
|
7 |
+
- master
|
8 |
+
jobs:
|
9 |
+
test:
|
10 |
+
runs-on: ubuntu-latest
|
11 |
+
|
12 |
+
steps:
|
13 |
+
- name: Checkout code
|
14 |
+
uses: actions/checkout@v2
|
15 |
+
|
16 |
+
- name: Set up Python
|
17 |
+
uses: actions/setup-python@v2
|
18 |
+
with:
|
19 |
+
python-version: 3.8
|
20 |
+
|
21 |
+
- name: Install dependencies
|
22 |
+
run: pip install -r requirements.txt # Adjust this based on your project structure
|
23 |
+
|
24 |
+
- name: Run tests
|
25 |
+
run: python -m unittest discover tests # Adjust this based on your test discovery method
|
README.md
CHANGED
@@ -31,17 +31,17 @@ ve-gans is a project for image generation using Generative Adversarial Networks
|
|
31 |
|
32 |
To train the GAN model, use the following command:
|
33 |
|
34 |
-
|
35 |
-
|
36 |
-
|
37 |
|
38 |
## Generating Images
|
39 |
|
40 |
To generate images with the trained model, use the following command:
|
41 |
|
42 |
-
|
43 |
-
|
44 |
-
|
45 |
|
46 |
## Project Structure
|
47 |
|
|
|
31 |
|
32 |
To train the GAN model, use the following command:
|
33 |
|
34 |
+
```bash
|
35 |
+
ve-gans-train
|
36 |
+
```
|
37 |
|
38 |
## Generating Images
|
39 |
|
40 |
To generate images with the trained model, use the following command:
|
41 |
|
42 |
+
```bash
|
43 |
+
ve-gans-generate
|
44 |
+
```
|
45 |
|
46 |
## Project Structure
|
47 |
|
todo.txt
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
TODO need to start working on comments
|
vegans/discriminator.py
CHANGED
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import torch.nn as nn
|
2 |
+
import torch.nn.functional as F
|
3 |
+
import torch
|
4 |
+
|
5 |
+
class Discriminator(nn.Module):
|
6 |
+
def __init__(self):
|
7 |
+
super(Discriminator, self).__init__()
|
8 |
+
|
9 |
+
def forward(self):
|
10 |
+
pass
|
vegans/generator.py
CHANGED
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import torch.nn as nn
|
2 |
+
import torch.nn.functional as F
|
3 |
+
import torch
|
4 |
+
|
5 |
+
class Generator(nn.Module):
|
6 |
+
def __init__(self):
|
7 |
+
super(Generator, self).__init__()
|
8 |
+
|
9 |
+
def forward(self):
|
10 |
+
pass
|
vegans/utils.py
CHANGED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
|
vegans/vegans.py
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
# initialize vegans
|