CarolXia commited on
Commit
7961c58
·
1 Parent(s): 8bb500b

single thread 50

Browse files
Files changed (1) hide show
  1. app.py +17 -17
app.py CHANGED
@@ -144,24 +144,24 @@ unmasked_text = dataset['unmasked_text'] # This will load the entire column inme
144
  st.write('Number of rows in the dataset ', dataset.num_rows)
145
  sizes = [0] * 5
146
  start = time.time()
147
- # t0 = threading.Thread(target=process_datasets, args=(0, 50, unmasked_text, sizes, 0, entity_set, []))
148
- # t1 = threading.Thread(target=process_datasets, args=(25, 50, unmasked_text, sizes, 1, entity_set, []))
149
- # t2 = threading.Thread(target=process_datasets, args=(20, 30, unmasked_text, sizes, 2, entity_set, []))
150
- # t3 = threading.Thread(target=process_datasets, args=(30, 40, unmasked_text, sizes, 3, entity_set, []))
151
- # t4 = threading.Thread(target=process_datasets, args=(40, 50, unmasked_text, sizes, 4, entity_set, []))
152
  # with profile(activities=[ProfilerActivity.CPU, ProfilerActivity.CUDA], profile_memory=True, record_shapes=True) as prof:
153
- process_datasets(0, 50, unmasked_text, sizes, 0, entity_set, [])
154
- # t0.start()
155
- # t1.start()
156
- # t2.start()
157
- # t3.start()
158
- # t4.start()
159
-
160
- # t0.join()
161
- # t1.join()
162
- # t2.join()
163
- # t3.join()
164
- # t4.join()
165
 
166
  end = time.time()
167
  length = end - start
 
144
  st.write('Number of rows in the dataset ', dataset.num_rows)
145
  sizes = [0] * 5
146
  start = time.time()
147
+ t0 = threading.Thread(target=process_datasets, args=(0, 10, unmasked_text, sizes, 0, entity_set, []))
148
+ t1 = threading.Thread(target=process_datasets, args=(10, 20, unmasked_text, sizes, 1, entity_set, []))
149
+ t2 = threading.Thread(target=process_datasets, args=(20, 30, unmasked_text, sizes, 2, entity_set, []))
150
+ t3 = threading.Thread(target=process_datasets, args=(30, 40, unmasked_text, sizes, 3, entity_set, []))
151
+ t4 = threading.Thread(target=process_datasets, args=(40, 50, unmasked_text, sizes, 4, entity_set, []))
152
  # with profile(activities=[ProfilerActivity.CPU, ProfilerActivity.CUDA], profile_memory=True, record_shapes=True) as prof:
153
+ # process_datasets(0, 50, unmasked_text, sizes, 0, entity_set, [])
154
+ t0.start()
155
+ t1.start()
156
+ t2.start()
157
+ t3.start()
158
+ t4.start()
159
+
160
+ t0.join()
161
+ t1.join()
162
+ t2.join()
163
+ t3.join()
164
+ t4.join()
165
 
166
  end = time.time()
167
  length = end - start