Skip to content
Snippets Groups Projects
Commit 45ff7def authored by tabetomo's avatar tabetomo Committed by Sanghyun Son
Browse files

fix multiprocessing issue in windows environment (#118)

parent 72f5cc26
No related branches found
No related tags found
1 merge request!1Jan 09, 2018 updates
...@@ -10,6 +10,8 @@ from trainer import Trainer ...@@ -10,6 +10,8 @@ from trainer import Trainer
torch.manual_seed(args.seed) torch.manual_seed(args.seed)
checkpoint = utility.checkpoint(args) checkpoint = utility.checkpoint(args)
def main():
global model
if args.data_test == 'video': if args.data_test == 'video':
from videotester import VideoTester from videotester import VideoTester
model = model.Model(args, checkpoint) model = model.Model(args, checkpoint)
...@@ -27,3 +29,5 @@ else: ...@@ -27,3 +29,5 @@ else:
checkpoint.done() checkpoint.done()
if __name__ == '__main__':
main()
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment