NOTE:Run this notebook in SageMaker instance instead of SageMaker Studio
paper: https://mal-net.org/ https://mal-net.org/#Citation
code: https://github.com/safreita1/malnet-graph THANKS
datasets: http://malnet.cc.gatech.edu/graph-data/
Core requirements:
- PyTorch1.13.1+py39
- extended pre-built container + BYOC
- SageMaker+Malnet-Graph
- BYOS
- using FastFile mode to access dataset files in S3
Steps:
- Generate Dockerfile
- Generate requirements.txt
- Build docker image
- SageMaker setting
- Download datasets and upload to s3
- Upload container image to Amazon ECR
- Start SageMaker training job
Please open the malnet-graph.ipynb file to walk through.