forked from apple/ml-kpconvx
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtest_ScanObjectNN.sh
More file actions
executable file
·39 lines (28 loc) · 903 Bytes
/
test_ScanObjectNN.sh
File metadata and controls
executable file
·39 lines (28 loc) · 903 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
#!/bin/bash
#
# For licensing see accompanying LICENSE file.
# Copyright (C) 2024 Apple Inc. All Rights Reserved.
#
# Go to the root directory
cd KPConvX
export PYTHONPATH=$PWD:$PYTHONPATH
############
# Parameters
############
# Change the path to your dataset here
ARGS="--dataset_path $PWD/../data/ScanObjectNN/main_split"
# Choose the log path here
LOG_PATH="$PWD/results/ScanObjectNN_KPConvD-L"
ARGS="$ARGS --log_path $LOG_PATH"
# # Optionally, you can choose a specific weight file
# ARGS="$ARGS --weight_path $LOG_PATH/checkpoints/current_chkp.tar"
# If you provide the weight path, it has to be in the log_path folder.
# It allows you to choose a specific weight file from the log folder.
###############
# Main function
###############
# Define the experiment and script to run
EXP="ScanObjectNN"
SCRIPT="test_ScanObj.py"
# Start the training
python3 experiments/$EXP/$SCRIPT $ARGS