Update train.py #260

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed

Nina1031 wants to merge 1 commit into CSAILVision:master from Nina1031:comment

Closed

Changes from all commits

Commits

Select commit

File filter

Conversations

Jump to

Jump to file

Failed to load files.

Update train.py

感觉 dataset_train 的cfg里选择了训练类别

Nina1031 authored May 25, 2021

commit 1cfd28074d1a5bee07a95b96d31d31b0152d893c

2 changes: 1 addition & 1 deletion train.py

Show comments View file Open in desktop

Original file line number	Diff line number	Diff line change
Expand Up		@@ -161,7 +161,7 @@ def main(cfg, gpus):
		net_encoder, net_decoder, crit)

		# Dataset and Loader
	dataset_train = TrainDataset(
	dataset_train = TrainDataset( # select trian index
		cfg.DATASET.root_dataset,
		cfg.DATASET.list_train,
		cfg.DATASET,
Expand Down

Navigation Menu