Enable Travis tests for adversarial_robustness

* Ensure that `pip` is up to date
* Remove invalid `arch` flag

PiperOrigin-RevId: 368226929
This commit is contained in:
Alistair Muldal
2021-04-13 17:07:51 +01:00
committed by Diego de Las Casas
parent 65e4ca602f
commit 76f6a815ff
2 changed files with 4 additions and 4 deletions
+1
View File
@@ -8,6 +8,7 @@ python:
- "3.6"
env:
matrix:
- PROJECT="adversarial_robustness"
- PROJECT="avae"
# - PROJECT="cs_gan" # TODO(b/184845450): Fix and re-enable
- PROJECT="iodine"
Regular → Executable
+3 -4
View File
@@ -13,20 +13,19 @@
# See the License for the specific language governing permissions and
# limitations under the License.
python3 -m venv adversarial_robustness_venv
source adversarial_robustness_venv/bin/activate
python3 -m venv /tmp/adversarial_robustness_venv
source /tmp/adversarial_robustness_venv/bin/activate
pip install -U pip
pip install -r adversarial_robustness/requirements.txt
python3 -m adversarial_robustness.jax.eval \
--ckpt=dummy \
--arch=wrn-10-1 \
--dataset=cifar10 \
--batch_size=1 \
--num_batches=1
python3 -m adversarial_robustness.pytorch.eval \
--ckpt=dummy \
--arch=wrn-10-1 \
--dataset=cifar10 \
--batch_size=1 \
--num_batches=1 \