From 4d78a9438016e6f53b86a40c5738bc7fd3e802b9 Mon Sep 17 00:00:00 2001 From: Ben Hoover Date: Thu, 8 Feb 2024 11:19:57 -0500 Subject: [PATCH 1/2] small update environment, readme, gitignore --- .gitignore | 3 +++ environment.yml | 10 ++++++++++ readme.md | 3 +++ 3 files changed, 16 insertions(+) create mode 100644 environment.yml diff --git a/.gitignore b/.gitignore index 3fc6050..e4cc048 100644 --- a/.gitignore +++ b/.gitignore @@ -1,3 +1,6 @@ +/data +results/ + *.tfevents* figures *.bak diff --git a/environment.yml b/environment.yml new file mode 100644 index 0000000..68b2f8c --- /dev/null +++ b/environment.yml @@ -0,0 +1,10 @@ +name: energy-transformer-torch +channels: + - pytorch + - conda-forge + - defaults + - anaconda +dependencies: + - python=3.10 + - pip>=20.0.0 + - ipykernel diff --git a/readme.md b/readme.md index 37bcf85..ea4d8ba 100644 --- a/readme.md +++ b/readme.md @@ -7,8 +7,11 @@ This is the official repository for the Energy Transformer (ET) written in PyTor # Installation ``` +conda env create -f environment.yml +conda activate energy-transformer-torch pip install -r requirements.txt ``` + If you encounter trouble with installing PyTorch, see https://pytorch.org for further instructions. Credits to [Ben Hoover](https://github.com/bhoov) for the diagrams.

From aed3137b4dc5e64e11d5ce70780f0ab94adb40b1 Mon Sep 17 00:00:00 2001 From: Ben Hoover Date: Thu, 8 Feb 2024 11:21:35 -0500 Subject: [PATCH 2/2] . --- readme.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/readme.md b/readme.md index ea4d8ba..3b02e9c 100644 --- a/readme.md +++ b/readme.md @@ -77,7 +77,7 @@ python train_mask_image_et.py \ --time_steps 12 \ --blocks 1 \ --epochs 1000 \ - --avg_gpus False \ + --avg_gpu False \ --result_path ./my_et \ --batch_size 128 \ --learning_rate 5e-5 \