From 134be1ac1775cc7cc2331f9d7e0d77aa330cdbbb Mon Sep 17 00:00:00 2001
From: Sungjin Ahn <55034857+agentml@users.noreply.github.com>
Date: Mon, 26 Dec 2022 10:25:16 +0900
Subject: [PATCH] Update README.md

---
 README.md | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/README.md b/README.md
index ec379ba..8a9c783 100644
--- a/README.md
+++ b/README.md
@@ -7,6 +7,9 @@ This is the official source code for _SLATE_. More details in the [paper](https:
 
 In this repository, we provide an implementation of the model, the training code and a dataset loader for the 3D Shapes dataset. This code is implemented in Pytorch.
 
+### Authors
+Gautam Singh and Fei Deng and Sungjin Ahn
+
 ### Dataset
 The current release provides code to train the model on the 3D Shapes dataset. The dataset class is provided in `shapes_3d.py`. You can edit or replace this class if you need to run the code on a different dataset. The 3D Shapes dataset can be downloaded from the [official URL](https://console.cloud.google.com/storage/browser/3d-shapes). This should produce a dataset file `3dshapes.h5`. During training, the path to this dataset file needs to be provided using the argument `--data_path`.