mirror of
https://github.com/google-deepmind/deepmind-research.git
synced 2026-05-09 21:07:49 +08:00
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:
committed by
Diego de Las Casas
parent
65e4ca602f
commit
76f6a815ff
@@ -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
@@ -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 \
|
||||
|
||||
Reference in New Issue
Block a user