Commit 00fc26a9 authored by Benjamin Vandersmissen's avatar Benjamin Vandersmissen
Browse files

updated Dockerfile and added additional parameter to jobRequest.json

parent 2b5536d2
......@@ -3,4 +3,4 @@ FROM pytorch/pytorch:latest
COPY requirements.txt /project/requirements.txt
RUN pip3 install -r /project/requirements.txt
COPY code /project/code
CMD python3 /project/code/main.py -d $LTH_DATA --lr $LTH_LR -n $LTH_N -p $LTH_P -b $LTH_B -t $LTH_T -r $LTH_R --device $LTH_DEVICE
CMD python3 /project/code/main.py -d $LTH_DATA --lr $LTH_LR -n $LTH_N -p $LTH_P -b $LTH_B -t $LTH_T -r $LTH_R --device $LTH_DEVICE -m $LTH_M
{
"name": "LTH test",
"name": "LTH test VGG11",
"deploymentEnvironment": "production",
"request": {
"resources": {
"cpus": 1,
"gpus": 1,
"cpuMemoryGb": 16,
"cpuMemoryGb": 32,
"minCudaVersion": 11,
"clusterId": 7
},
"docker": {
"image": "gitlab+deploy-token-535:19ESJYmR2qaQn5toyB8s@gitlab.ilabt.imec.be:4567/sparse-representation-learning/lth:latest",
"environment":{
"LTH_DATA": "fashion",
"LTH_LR": 0.0005,
"LTH_N": 20,
"LTH_DATA": "flowers",
"LTH_LR": 0.00025,
"LTH_N": 25,
"LTH_P": 0.2,
"LTH_B": 128,
"LTH_T":15,
"LTH_R":42,
"LTH_DEVICE":"cuda:0"
"LTH_B": 32,
"LTH_T": 15,
"LTH_R": 42,
"LTH_DEVICE":"cuda:0",
"LTH_M": "vgg"
},
"storage": [
{
......@@ -29,5 +30,5 @@
]
}
},
"description": "testing the LTH project"
"description": "testing the implementation & pruning of VGG11"
}
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment