Commit 9fdc006f authored by Benjamin Vandersmissen's avatar Benjamin Vandersmissen
Browse files

Updated the Dockerfile and the request files for GPULab

parent 87c962a1
......@@ -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 -m $LTH_M
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 -o $LTH_O -i $LTH_I --initialization $LTH_INITIALIZATION
{
"name": "LTH VGG11",
"name": "LTH VGG",
"deploymentEnvironment": "production",
"request": {
"resources": {
"cpus": 1,
"cpus": 2,
"gpus": 1,
"cpuMemoryGb": 32,
"minCudaVersion": 11,
......@@ -12,15 +12,18 @@
"docker": {
"image": "gitlab+deploy-token-535:19ESJYmR2qaQn5toyB8s@gitlab.ilabt.imec.be:4567/sparse-representation-learning/lth:latest",
"environment":{
"LTH_DATA": "imagenette",
"LTH_LR": 0.0005,
"LTH_DATA": "cifar10",
"LTH_LR": 0.00025,
"LTH_N": 20,
"LTH_P": 0.2,
"LTH_B": 64,
"LTH_T": 2,
"LTH_T": 40,
"LTH_R": 42,
"LTH_DEVICE":"cuda:0",
"LTH_M": "vgg"
"LTH_M": "vgg",
"LTH_O": 0,
"LTH_I": "none",
"LTH_INITIALIZATION": "kaiming_uniform"
},
"storage": [
{
......@@ -30,5 +33,5 @@
]
}
},
"description": "testing the implementation & pruning of VGG11"
"description": "testing the implementation & pruning of VGG"
}
{
"name": "LTH test",
"name": "LTH file transfer",
"deploymentEnvironment": "production",
"request": {
"resources": {
......@@ -11,16 +11,6 @@
},
"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": 10,
"LTH_P": 0.2,
"LTH_B": 32,
"LTH_T":10,
"LTH_R":42,
"LTH_DEVICE":"cuda:0"
},
"storage": [
{
"containerPath": "/project",
......@@ -30,5 +20,5 @@
"command": "bash -c 'apt-get update; apt-get install -y rsync vim; sleep 36000'"
}
},
"description": "transfering files to the LTH project"
"description": "transferring files to the LTH project"
}
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