Commit a3b4fbda authored by Marco Di Francesco's avatar Marco Di Francesco 🍉
Browse files

Update final taasss

parent 5a69648d
/data1/IDA_LSTM_checkpoints
\ No newline at end of file
......@@ -26,7 +26,7 @@ parser.add_argument("--n_gpu", type=int, default=0)
parser.add_argument("--dataset_name", type=str, default="radar")
parser.add_argument("--r", type=int, default=4)
parser.add_argument("--is_parallel", type=bool, default=False)
parser.add_argument("--save_dir", type=str, default="/data1/IDA_LSTM_checkpoints")
parser.add_argument("--save_dir", type=str, default="checkpoints/")
parser.add_argument("--gen_frm_dir", type=str, default="dataset_generated/")
parser.add_argument("--input_length", type=int, default=5)
parser.add_argument("--total_length", type=int, default=25) # 15
......@@ -61,12 +61,12 @@ parser.add_argument("--num_save_samples", type=int, default=10)
args = parser.parse_args()
batch_size = args.batch_size
# if args.img_width == 128:
# args.patch_size = 4
# elif args.img_width == 256:
# args.patch_size = 8
# elif args.img_width == 512:
# args.patch_size = 16
if args.img_width == 128:
args.patch_size = 4
elif args.img_width == 256:
args.patch_size = 8
elif args.img_width == 512:
args.patch_size = 16
def seed_everything(seed):
......
This source diff could not be displayed because it is too large. You can view the blob instead.
......@@ -9,13 +9,22 @@ from data_provider.CIKM.taasss import infinite_batcher
def get_batcher(args):
# data_dir = Path("/") / "home" / "gabriele" / "Documents" / "dottorato" / "data" / "meteotn_data_new"
# metadata_file = data_dir / "hdf_metadata.csv"
data_dir = Path("/") / "data1" / "meteotn_data_2010_2016"
metadata_file = data_dir / "run_metadata.csv"
# data_dir = Path("/") / "data1" / "meteotn_data_2010_2016"
# metadata_file = data_dir / "run_metadata.csv"
# all_data = h5py.File(
# data_dir / "hdf_archives" / "all_data.hdf5",
# "r",
# libver="latest",
# )
data_dir = Path("/data2/franch")
metadata_file = data_dir / "hdf_metadata.csv"
all_data = h5py.File(
data_dir / "hdf_archives" / "all_data.hdf5",
"r",
libver="latest",
data_dir / "hdf_archives" / "all_data.hdf5", "r", libver="latest"
)
mask_path = data_dir / "mask.png"
assert mask_path.exists(), "Mask does not exist"
outlier_mask = cv2.imread(str(mask_path), 0)
......
Supports Markdown
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