Skip to content
Snippets Groups Projects
Unverified Commit 8a7a40cb authored by i-robot's avatar i-robot Committed by Gitee
Browse files

!2281 add dataset link for naml & fix unet bug

Merge pull request !2281 from zhaoting/master
parents cc249a8a 129513ab
No related branches found
No related tags found
No related merge requests found
......@@ -89,7 +89,9 @@ def train_net(cross_valid_ind=1,
eval_resize=config.eval_resize, split=split, shuffle=False)
else:
repeat = config.repeat
dataset_sink_mode = True
dataset_sink_mode = False
if config.device_target == "GPU":
dataset_sink_mode = True
per_print_times = 1
train_dataset, valid_dataset = create_dataset(data_dir, repeat, batch_size, True, cross_valid_ind,
run_distribute, config.crop, config.image_size)
......
......@@ -22,7 +22,9 @@ NAML is a multi-view news recommendation approach. The core of NAML is a news en
# [Dataset](#contents)
Dataset used: [MIND](https://msnews.github.io/)
Dataset used: [MIND](https://msnews.github.io/). You can download [MINDlarge_train](https://mind201910small.blob.core.windows.net/release/MINDlarge_train.zip),
[MINDlarge_dev](https://mind201910small.blob.core.windows.net/release/MINDlarge_dev.zip),
[MINDlarge_utils](https://mind201910small.blob.core.windows.net/release/MINDlarge_utils.zip)
MIND contains about 160k English news articles and more than 15 million impression logs generated by 1 million users.
......
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