Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
@@ -1,3 +1,6 @@
/data
results/

*.tfevents*
figures
*.bak
Expand Down
10 changes: 10 additions & 0 deletions environment.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
name: energy-transformer-torch
channels:
- pytorch
- conda-forge
- defaults
- anaconda
dependencies:
- python=3.10
- pip>=20.0.0
- ipykernel
5 changes: 4 additions & 1 deletion readme.md
Original file line number Diff line number Diff line change
Expand Up @@ -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.

<p float="center">
Expand Down Expand Up @@ -74,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 \
Expand Down