examples/milopyp.md
... ...
@@ -29,6 +29,9 @@ mv *.rec *.ali *.tlt ./sample_data
29 29
30 30
## -- Cellular content exploration module --
31 31
32
+## Specify GPU to use - this must be set
33
+export CUDA_VISIBLE_DEVICES=0
34
+
32 35
## Training 2d3d
33 36
python.milopyp ${milo_dir}/simsiam_main.py simsiam2d3d --num_epochs 20 --exp_id test_sample --bbox 36 --dataset simsiam2d3d --arch simsiam2d3d_18 --lr 1e-3 --train_img_txt sample_train_explore_img.txt --batch_size 256 --val_intervals 20 --save_all --gauss 0.8 --dog 3,5
34 37